Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h440 int64_t dim1_size = values.size(); in CreateR3Projected() local
465 int64_t dim1_size = projection_z; in CreateR4Projected() local
Dliteral.h1330 const int64_t dim1_size = values.begin()->size(); in PopulateR2() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc57 const std::string& tensor_name, int dim1_size, in CopySubArrayToArray()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc183 int dim1_size = 1; in MaybeMergeConstNodes() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc277 const int64_t dim1_size = 1024; in XLA_TEST_F() local