Home
last modified time | relevance | path

Searched defs:dim1_size (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/protobuf/tpu/
Dtpu_embedding_output_layout.proto63 int32 dim1_size = 1; field
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h481 int64 dim1_size = values.size(); in CreateR3Projected() local
506 int64 dim1_size = projection_z; in CreateR4Projected() local
Dliteral.h1010 const int64 dim1_size = values.begin()->size(); in PopulateR2() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc55 const string& tensor_name, int dim1_size, in CopySubArrayToArray()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc182 int dim1_size = 1; in MaybeMergeConstNodes() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc280 const int64 dim1_size = 1024; in XLA_TEST_F() local