Searched defs:dim_sizes (Results 1 – 11 of 11) sorted by relevance
105 def testBroadcastToRank(self, dim_sizes, rank, expected_dim_sizes): argument413 def testRaggedBroadcastTo(self, x, dim_sizes, expected): argument
152 def from_dim_sizes(dim_sizes): argument
180 auto dim_sizes = input_shape.dim_sizes(); in Compile() local
157 TensorShapeBase<Shape>::TensorShapeBase(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeBase()180 void TensorShapeBase<Shape>::InitDims(gtl::ArraySlice<int64> dim_sizes) { in InitDims()424 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::__anon84b1e2d30111::TensorShapeOld390 TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeOld()
171 TensorShapeBase(std::initializer_list<int64> dim_sizes) in TensorShapeBase()
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()
169 std::vector<int64> dim_sizes(num_outer_dims_to_partition); in TEST_F() local
122 const SparsityParameters& sparsity, const std::vector<int>& dim_sizes) { in VerifyAndCountElements()
240 auto shape_proto = [](std::initializer_list<int64> dim_sizes) { in TEST()
132 string dim_sizes, indices; in AddRewritesForShape() local