Searched defs:dim_sizes (Results 1 – 12 of 12) sorted by relevance
112 def testBroadcastToRank(self, dim_sizes, rank, expected_dim_sizes): argument420 def testRaggedBroadcastTo(self, x, dim_sizes, expected): argument
132 def from_dim_sizes(dim_sizes): argument
68 const auto dim_sizes = then_shape.dim_sizes(); in Compile() local
180 auto dim_sizes = input_shape.dim_sizes(); in Compile() local
136 TensorShapeBase<Shape>::TensorShapeBase(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeBase()159 void TensorShapeBase<Shape>::InitDims(gtl::ArraySlice<int64> dim_sizes) { in InitDims()403 gtl::InlinedVector<int64, 4> TensorShapeBase<Shape>::dim_sizes() const { in dim_sizes() function in tensorflow::TensorShapeBase
221 TensorShapeOld(std::initializer_list<int64> dim_sizes) in TensorShapeOld()276 gtl::ArraySlice<int64> dim_sizes() const { return dim_sizes_; } in dim_sizes() function in tensorflow::__anone850e9110111::TensorShapeOld390 TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeOld()
171 TensorShapeBase(std::initializer_list<int64> dim_sizes) in TensorShapeBase()
1610 auto make_shape = [&c](std::initializer_list<int64> dim_sizes) { in TestMergeHandles()1721 auto make_shape = [&c](std::initializer_list<int64> dim_sizes) { in TestRelaxHandles()
169 std::vector<int64> dim_sizes(num_outer_dims_to_partition); in TEST_F() local
219 auto shape_proto = [](std::initializer_list<int64> dim_sizes) { in TEST()
167 def _IotaNdF32Constant(dim_sizes): argument
131 string dim_sizes, indices; in AddRewritesForShape() local