| /external/tensorflow/tensorflow/c/eager/ |
| D | unified_api_test.cc | 93 int64 dim_sizes[] = {2, 4}; in TestTensorShape2x4() local
122 int64_t dim_sizes[] = {2, 4}; in TEST_P() local
173 int64 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 | 105 def testBroadcastToRank(self, dim_sizes, rank, expected_dim_sizes): argument
416 def testRaggedBroadcastTo(self, x, dim_sizes, expected): argument
|
| D | ragged_tensor_shape.py | 152 def from_dim_sizes(dim_sizes): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.cc | 184 TensorShapeBase<Shape>::TensorShapeBase(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeBase()
192 gtl::ArraySlice<int64> dim_sizes, TensorShapeBase* out) { in BuildTensorShapeBase()
213 Status TensorShapeBase<Shape>::InitDims(gtl::ArraySlice<int64> dim_sizes) { in InitDims()
557 gtl::InlinedVector<int64, 4> TensorShapeBase<Shape>::dim_sizes() const { in dim_sizes() function in tensorflow::TensorShapeBase
|
| D | tensor_shape.h | 171 TensorShapeBase(std::initializer_list<int64> dim_sizes) in TensorShapeBase()
186 static Status BuildTensorShapeBase(std::initializer_list<int64> dim_sizes, in BuildTensorShapeBase()
|
| D | tensor_shape_test.cc | 244 TensorShapeOld(std::initializer_list<int64> dim_sizes) in TensorShapeOld()
299 gtl::ArraySlice<int64> dim_sizes() const { return dim_sizes_; } in dim_sizes() function in tensorflow::__anon127dd62f0111::TensorShapeOld
413 TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeOld()
|
| D | shape_inference_test.cc | 1609 auto make_shape = [&c](std::initializer_list<int64> dim_sizes) { in TestMergeHandles()
1720 auto make_shape = [&c](std::initializer_list<int64> dim_sizes) { in TestRelaxHandles()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.cc | 49 gtl::InlinedVector<int64, 4> dim_sizes = shape.dim_sizes(); in CreateUninitializedResourceVariable() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 184 auto dim_sizes = input_shape.dim_sizes(); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition_test.cc | 169 std::vector<int64> dim_sizes(num_outer_dims_to_partition); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 237 const SparsityParameters& sparsity, const std::vector<int>& dim_sizes) { in VerifyAndCountElements()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 240 auto shape_proto = [](std::initializer_list<int64> dim_sizes) { in TEST()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 463 auto dim_sizes = first_shape.dim_sizes(); in Shape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 724 absl::Span<HloInstruction* const> dim_sizes() const { in dim_sizes() function
733 HloInstruction* dim_sizes(int64 i) const { return operands()[i + 1]; } in dim_sizes() function
|
| D | hlo_instruction.cc | 1460 absl::Span<HloInstruction* const> dim_sizes) { in CreateDynamicReshape()
|
| D | hlo_instructions.cc | 1083 absl::Span<HloInstruction* const> dim_sizes) in HloDynamicReshapeInstruction()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen.cc | 133 string dim_sizes, indices; in AddRewritesForShape() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | rebatch_dataset_op.cc | 460 auto dim_sizes = dataset()->output_shapes()[i].dim_sizes(); in GetNextInternal() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 168 DataType dtype, absl::Span<const int64> dim_sizes) { in CreateTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1148 absl::Span<const XlaOp> dim_sizes, in DynamicReshape()
4137 XlaOp DynamicReshape(XlaOp operand, absl::Span<const XlaOp> dim_sizes, in DynamicReshape()
|