Searched refs:dim1_size (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | lstm_utils.cc | 55 const string& tensor_name, int dim1_size, in CopySubArrayToArray() argument
60 Shape shape = is_bias ? Shape({dim1_size}) : Shape({dim1_size, dim2_size}); in CopySubArrayToArray()
67 start_idx1, start_idx2, buffer, dim2_size, 0, 0, dim1_size, in CopySubArrayToArray()
|
| D | lstm_utils.h | 95 const string& tensor_name, int dim1_size,
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf.cc | 182 int dim1_size = 1; in MaybeMergeConstNodes() local
205 dim1_size *= shape.dim(d).size(); in MaybeMergeConstNodes()
230 new float[dim0_size * dim1_size]); in MaybeMergeConstNodes()
232 dim0_size, dim1_size, transposed_tensor.get()); in MaybeMergeConstNodes()
234 allocated_tensor_shape->add_dim()->set_size(dim1_size); in MaybeMergeConstNodes()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 481 int64 dim1_size = values.size(); in CreateR3Projected() local
484 Array3D<NativeT> array(dim0_size, dim1_size, dim2_size); in CreateR3Projected()
496 CHECK_EQ(dim1_size, dim1); in CreateR3Projected()
506 int64 dim1_size = projection_z; in CreateR4Projected() local
510 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size); in CreateR4Projected()
512 for (int64 dim1 = 0; dim1 < dim1_size; ++dim1) { in CreateR4Projected()
|
| D | literal.h | 1010 const int64 dim1_size = values.begin()->size(); in PopulateR2() local
1012 CHECK_EQ(dim1_size, shape().dimensions(1)); in PopulateR2()
1021 CHECK_EQ(dim1_size, dim1); in PopulateR2()
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_output_layout.proto | 29 // layout.output(tensor_index).dim1_size() - table width; repeated or
63 int32 dim1_size = 1; field
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 77 dim1->set_size(output.two_d().dim1_size()); in ComputeOutputTensorShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 280 const int64 dim1_size = 1024; in XLA_TEST_F() local
282 ShapeUtil::MakeShapeWithLayout(F32, {rand_dim0_size, dim1_size}, {1, 0}); in XLA_TEST_F()
301 for (int j = 0; j < dim1_size; ++j) { in XLA_TEST_F()
|