Home
last modified time | relevance | path

Searched refs:dim0_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc181 int dim0_size = 0; in MaybeMergeConstNodes() local
200 dim0_size += shape.dim(0).size(); in MaybeMergeConstNodes()
230 new float[dim0_size * dim1_size]); in MaybeMergeConstNodes()
232 dim0_size, dim1_size, transposed_tensor.get()); in MaybeMergeConstNodes()
235 allocated_tensor_shape->add_dim()->set_size(dim0_size); in MaybeMergeConstNodes()
242 tensor_shape_dim0->set_size(dim0_size); in MaybeMergeConstNodes()
/external/tensorflow/tensorflow/core/framework/
Dops_util.h51 const int64 dim0_size = s.dim_size(0); in IsInnerDimsSizeAligned() local
52 if (dim0_size == 0) return false; in IsInnerDimsSizeAligned()
56 const int64 bytes_per_dim0 = (s.num_elements() / dim0_size) * sizeof(T); in IsInnerDimsSizeAligned()
Dtensor.cc809 int64 dim0_size = shape_.dim_size(0); in Slice() local
810 CHECK_LE(limit, dim0_size); in Slice()
811 if ((start == 0) && (limit == dim0_size)) { in Slice()
818 if (dim0_size > 0) { in Slice()
819 const int64 elems_per_dim0 = NumElements() / dim0_size; in Slice()
821 dim0_size = limit - start; in Slice()
822 ret.shape_.set_dim(0, dim0_size); in Slice()
823 const int64 num_elems = dim0_size * elems_per_dim0; in Slice()
835 int64 dim0_size = shape_.dim_size(0); in SubSlice() local
836 CHECK_LE(index, dim0_size); // Crash ok. in SubSlice()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h480 int64 dim0_size = projection; in CreateR3Projected() local
484 Array3D<NativeT> array(dim0_size, dim1_size, dim2_size); in CreateR3Projected()
485 for (int64 dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR3Projected()
505 int64 dim0_size = projection_p; in CreateR4Projected() local
510 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size); in CreateR4Projected()
511 for (int64 dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR4Projected()
Dliteral.h1009 const int64 dim0_size = values.size(); in PopulateR2() local
1011 CHECK_EQ(dim0_size, shape().dimensions(0)); in PopulateR2()