Home
last modified time | relevance | path

Searched refs:dynamic_dimensions (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape.h43 absl::Span<const bool> dynamic_dimensions, in Shape() argument
47 dynamic_dimensions_(dynamic_dimensions.begin(), in Shape()
48 dynamic_dimensions.end()), in Shape()
87 absl::Span<const bool> dynamic_dimensions() const { in dynamic_dimensions() function
Dshape_util.cc265 const std::vector<bool>& dynamic_dimensions) { in MakeShape() argument
266 return MakeValidatedShape(element_type, dimensions, dynamic_dimensions) in MakeShape()
290 const std::vector<bool>& dynamic_dimensions) { in MakeValidatedShape() argument
291 if (dynamic_dimensions.size() != dimensions.size()) { in MakeValidatedShape()
294 dynamic_dimensions.size(), dimensions.size()); in MakeValidatedShape()
304 shape.set_dynamic_dimension(i, dynamic_dimensions[i]); in MakeValidatedShape()
Dshape_util.h440 const std::vector<bool>& dynamic_dimensions);
449 const std::vector<bool>& dynamic_dimensions);
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc666 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
688 ContainersEqual(result_shape.dynamic_dimensions(), {false, true, false})) in TEST_F()
709 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
732 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true})) in TEST_F()
760 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
801 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), in TEST_F()
841 ContainersEqual(result_shape.dynamic_dimensions(), {true, true, false})) in TEST_F()
866 EXPECT_TRUE(ContainersEqual(result_shape.dynamic_dimensions(), {true, false})) in TEST_F()
895 ContainersEqual(result_shape.dynamic_dimensions(), {true, false, false})) in TEST_F()
931 EXPECT_TRUE(ContainersEqual(result_shape.tuple_shapes(0).dynamic_dimensions(), in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc120 std::vector<bool> dynamic_dimensions(rank, false); in TensorShapeToXLAShape() local
125 dynamic_dimensions[d] = true; in TensorShapeToXLAShape()
139 result.set_dynamic_dimension(d, dynamic_dimensions[d]); in TensorShapeToXLAShape()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc216 CopyVector(xla_shape.dynamic_dimensions(), &c_shape->dynamic_dimensions); in ToC()
236 absl::Span<const bool> dynamic_dims = MakeSpan(c_shape->dynamic_dimensions); in FromC()
256 if (c_shape->dynamic_dimensions.size > TPU_C_API_MAX_INLINED) { in Free()
257 delete[] c_shape->dynamic_dimensions.heap; in Free()
Dc_api_decl.h222 BoolList dynamic_dimensions; member
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dtransform_unranked_hlo.cc380 auto dynamic_dimensions = llvm::SmallVector<int64_t, 6>( in createRankSpecializedBroadcastAndOp() local
383 dynamic_dimensions, in createRankSpecializedBroadcastAndOp()
399 RankedTensorType::get(dynamic_dimensions, result_element_type); in createRankSpecializedBroadcastAndOp()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc448 std::vector<bool>* dynamic_dimensions);
4417 std::vector<bool>* dynamic_dimensions) { in ParseDimensionSizes() argument
4429 dynamic_dimensions->push_back(is_dynamic); in ParseDimensionSizes()
4595 std::vector<bool> dynamic_dimensions; in ParseShape() local
4596 if (!ParseDimensionSizes(&dimension_sizes, &dynamic_dimensions)) { in ParseShape()
4602 result->set_dynamic_dimension(i, dynamic_dimensions[i]); in ParseShape()
Dshape_inference.cc3341 start_indices_shape.dynamic_dimensions(), in InferGatherShape()