| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.h | 169 TensorShapeBase(std::initializer_list<int64_t> dim_sizes) in TensorShapeBase()
185 static Status BuildTensorShapeBase(std::initializer_list<int64_t> dim_sizes, in BuildTensorShapeBase()
366 static Status BuildTensorShape(gtl::ArraySlice<int64_t> dim_sizes, in BuildTensorShape()
370 static Status BuildTensorShape(std::initializer_list<int64_t> dim_sizes, in BuildTensorShape()
541 static Status BuildPartialTensorShape(gtl::ArraySlice<int64_t> dim_sizes, in BuildPartialTensorShape()
546 std::initializer_list<int64_t> dim_sizes, PartialTensorShape* out) { in BuildPartialTensorShape()
|
| D | tensor_shape.cc | 183 TensorShapeBase<Shape>::TensorShapeBase(gtl::ArraySlice<int64_t> dim_sizes) { in TensorShapeBase()
191 gtl::ArraySlice<int64_t> dim_sizes, TensorShapeBase* out) { in BuildTensorShapeBase()
212 Status TensorShapeBase<Shape>::InitDims(gtl::ArraySlice<int64_t> dim_sizes) { in InitDims()
559 gtl::InlinedVector<int64_t, 4> TensorShapeBase<Shape>::dim_sizes() const { in dim_sizes() function in tensorflow::TensorShapeBase
|
| D | tensor_shape_test.cc | 244 TensorShapeOld(std::initializer_list<int64_t> dim_sizes) in TensorShapeOld()
299 gtl::ArraySlice<int64_t> dim_sizes() const { return dim_sizes_; } in dim_sizes() function in tensorflow::__anon3caac34d0111::TensorShapeOld
413 TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64_t> dim_sizes) { in TensorShapeOld()
|
| D | shape_inference_test.cc | 1709 auto make_shape = [&c](std::initializer_list<int64_t> dim_sizes) { in TEST_P()
1813 auto make_shape = [&c](std::initializer_list<int64_t> dim_sizes) { in TEST_P()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | unified_api_test.cc | 93 int64_t dim_sizes[] = {2, 4}; in TestTensorShape2x4() local
122 int64_t dim_sizes[] = {2, 4}; in TEST_P() local
173 int64_t dim_sizes[] = {2, -1}; in TEST_P() local
|
| D | c_api_unified_experimental.h | 70 int64_t* dim_sizes; member
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape_test.py | 101 def testBroadcastToRank(self, dim_sizes, rank, expected_dim_sizes): argument
412 def testRaggedBroadcastTo(self, x, dim_sizes, expected): argument
|
| D | ragged_tensor_shape.py | 148 def from_dim_sizes(dim_sizes): argument
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | tensor_util.h | 60 auto dim_sizes = tf_tensor.shape().dim_sizes(); in GetTensorMetadata() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cross_op.cc | 52 const auto& dim_sizes = in0_shape.dim_sizes(); in Compile() local
|
| D | split_op.cc | 188 auto dim_sizes = input_shape.dim_sizes(); in Compile() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.cc | 47 gtl::InlinedVector<int64_t, 4> dim_sizes = shape.dim_sizes(); in CreateUninitializedResourceVariable() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition_test.cc | 169 std::vector<int64_t> dim_sizes(num_outer_dims_to_partition); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 240 auto shape_proto = [](std::initializer_list<int64_t> dim_sizes) { in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 229 const SparsityParameters& sparsity, const std::vector<int>& dim_sizes) { in VerifyAndCountElements()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen.cc | 133 string dim_sizes, indices; in AddRewritesForShape() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 591 auto dim_sizes = combined_shape.dim_sizes(); in Shape() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | tensor_layout.cc | 319 std::vector<int64_t> Mesh::dim_sizes() const { in dim_sizes() function in tensorflow::dtensor::Mesh
320 std::vector<int64_t> dim_sizes; in dim_sizes() local
|
| D | dtensor_device.cc | 1717 const auto dim_sizes = function.local_output_shapes.at(i).dim_sizes(); in ExecuteRegularOperation() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | rebatch_dataset_op.cc | 473 auto dim_sizes = dataset()->output_shapes()[i].dim_sizes(); in GetNextInternal() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 971 absl::Span<HloInstruction* const> dim_sizes() const { in dim_sizes() function
980 HloInstruction* dim_sizes(int64_t i) const { return operands()[i + 1]; } in dim_sizes() function
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | op_handler_selector_test.cc | 118 int64_t dim_sizes[] = {1}; in Shape() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 111 TensorMetadata CreateMetadata(DType dtype, absl::Span<const Index> dim_sizes) { in CreateMetadata()
687 tensorflow::DataType dtype, absl::Span<const int64_t> dim_sizes) { in CreateTensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 249 DataType dtype, absl::Span<const int64_t> dim_sizes) { in CreateTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1215 absl::Span<const XlaOp> dim_sizes, in DynamicReshape()
4093 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes, in DynamicReshape()
|