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.proto67 int32 dim1_size = 1; field
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h448 int64 dim1_size = values.size(); in CreateR3Projected() local
473 int64 dim1_size = projection_z; in CreateR4Projected() local
Dliteral.h1012 const int64 dim1_size = values.begin()->size(); in PopulateR2() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc55 const std::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/
Dcpu_gpu_fusion_test.cc279 const int64 dim1_size = 1024; in XLA_TEST_F() local