Searched defs:dynamic_dimensions (Results 1 – 9 of 9) sorted by relevance
156 std::optional<const std::vector<bool>> dynamic_dimensions) { in MakeShapeWithLayout()270 std::optional<std::vector<bool>> dynamic_dimensions) in BuildXlaCompilerSubmodule()271 -> StatusOr<Shape> { in BuildXlaCompilerSubmodule()
43 absl::Span<const bool> dynamic_dimensions, in Shape()91 absl::Span<const bool> dynamic_dimensions() const { in dynamic_dimensions() function
268 const std::vector<bool>& dynamic_dimensions) { in MakeShape()293 const std::vector<bool>& dynamic_dimensions) { in MakeValidatedShape()
212 std::vector<int> dynamic_dimensions; member
2169 std::vector<int>& dynamic_dimensions) { in AppendDynamicDimensions()2200 std::vector<int> dynamic_dimensions; in CreateExecutionCacheSignature() local
164 std::vector<bool> dynamic_dimensions(rank, false); in CompileToCustomCallCallingTfKernel() local
248 BoolList dynamic_dimensions; member
4991 std::vector<bool>* dynamic_dimensions) { in ParseDimensionSizes()5214 std::vector<bool> dynamic_dimensions; in ParseShape() local
641 BoolList dynamic_dimensions; // out member