Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dunroll_batch_matmul.cc50 CreateInt32Array(model, absl::StrCat(base_name, "/reshape_a/shape"), in SliceInput()
67 CreateInt32Array(model, absl::StrCat(batch_name, "/slice/begin"), in SliceInput()
69 CreateInt32Array(model, absl::StrCat(batch_name, "/slice/size"), in SliceInput()
81 CreateInt32Array(model, absl::StrCat(batch_name, "/reshape/shape"), in SliceInput()
119 const std::string perm_array_name = CreateInt32Array( in TransposeInput()
259 CreateInt32Array(model, base_name + "/reshape_out/shape", result_shape)}; in Run()
H A Dresolve_tensorflow_matmul.cc112 CreateInt32Array( in Run()
147 CreateInt32Array( in Run()
H A Dconvert_squeeze_to_reshape.cc71 CreateInt32Array(model, squeeze_op->outputs[0] + "_shape", in Run()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.h267 std::string CreateInt32Array(Model* model, const std::string& param_name,
H A Dtooling_util.cc1401 CreateInt32Array( in InsertCopyOperator()
1884 std::string CreateInt32Array(Model* model, const std::string& param_name, in CreateInt32Array() function