Home
last modified time | relevance | path

Searched defs:dim_sizes (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape.h169 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()
H A Dtensor_shape.cc183 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
H A Dtensor_shape_test.cc244 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::__anon86ba443e0111::TensorShapeOld
413 TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64_t> dim_sizes) { in TensorShapeOld()
/aosp_15_r20/external/federated-compute/fcp/aggregation/core/
H A Dtensor_shape.h45 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()
H A Dtensor_shape.cc42 TensorShape::DimSizesVector dim_sizes; in FromProto() local
H A Dtensor.proto49 repeated int64 dim_sizes = 1; field
H A Dtensor_shape_test.cc34 std::vector<size_t> dim_sizes = {4, 8, 3, 2}; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dunified_api_test.cc93 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
H A Dc_api_unified_experimental.h70 int64_t* dim_sizes; member
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_shape_test.py101 def testBroadcastToRank(self, dim_sizes, rank, expected_dim_sizes): argument
412 def testRaggedBroadcastTo(self, x, dim_sizes, expected): argument
H A Dragged_tensor_shape.py148 def from_dim_sizes(dim_sizes): argument
/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dconverters_test.cc43 tf::TensorShape CreateTfShape(std::initializer_list<int64_t> dim_sizes) { in CreateTfShape()
51 std::initializer_list<int64_t> dim_sizes) { in CreateTfTensorSpec()
H A Dconverters.cc55 std::vector<size_t> dim_sizes; in ConvertShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dtensor_util.h60 auto dim_sizes = tf_tensor.shape().dim_sizes(); in GetTensorMetadata() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dcross_op.cc52 const auto& dim_sizes = in0_shape.dim_sizes(); in Compile() local
H A Dsplit_op.cc188 auto dim_sizes = input_shape.dim_sizes(); in Compile() local
/aosp_15_r20/external/federated-compute/fcp/aggregation/testing/
H A Dtesting.cc48 tf::Tensor CreateStringTfTensor(std::initializer_list<int64_t> dim_sizes, in CreateStringTfTensor()
H A Dtesting.h40 std::initializer_list<int64_t> dim_sizes, in CreateTfTensor()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
H A Dvariable_ops.cc47 gtl::InlinedVector<int64_t, 4> dim_sizes = shape.dim_sizes(); in CreateUninitializedResourceVariable() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dshape_partition_test.cc169 std::vector<int64_t> dim_sizes(num_outer_dims_to_partition); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dtensor_layout.cc319 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instructions.h971 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
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops_test.cc240 auto shape_proto = [](std::initializer_list<int64_t> dim_sizes) { in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier.cc229 const SparsityParameters& sparsity, const std::vector<int>& dim_sizes) { in VerifyAndCountElements()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/
H A Dcodegen.cc133 string dim_sizes, indices; in AddRewritesForShape() local

12