| /external/federated-compute/fcp/aggregation/core/ |
| D | tensor_shape.h | 45 TensorShape(std::initializer_list<size_t> dim_sizes) in TensorShape()
58 const DimSizesVector& dim_sizes() const { return dim_sizes_; } in dim_sizes() function
74 explicit TensorShape(DimSizesVector&& dim_sizes) in TensorShape()
|
| D | tensor_shape.cc | 42 TensorShape::DimSizesVector dim_sizes; in FromProto() local
|
| D | tensor.proto | 49 repeated int64 dim_sizes = 1; field
|
| D | tensor_shape_test.cc | 34 std::vector<size_t> dim_sizes = {4, 8, 3, 2}; in TEST() local
|
| /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::__anon6c09febe0111::TensorShapeOld
413 TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64_t> dim_sizes) { in TensorShapeOld()
|
| /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/federated-compute/fcp/aggregation/tensorflow/ |
| D | converters_test.cc | 43 tf::TensorShape CreateTfShape(std::initializer_list<int64_t> dim_sizes) { in CreateTfShape()
51 std::initializer_list<int64_t> dim_sizes) { in CreateTfTensorSpec()
|
| D | converters.cc | 55 std::vector<size_t> dim_sizes; in ConvertShape() local
|
| /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/federated-compute/fcp/aggregation/testing/ |
| D | testing.cc | 48 tf::Tensor CreateStringTfTensor(std::initializer_list<int64_t> dim_sizes, in CreateStringTfTensor()
|
| D | testing.h | 40 std::initializer_list<int64_t> dim_sizes, in CreateTfTensor()
|
| /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
|