Searched refs:CreateInt32Array (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | unroll_batch_matmul.cc | 53 CreateInt32Array(model, batch_name + "/slice_a/slice/begin", in UnrollBatchMatMul3D() 55 CreateInt32Array(model, batch_name + "/slice_a/slice/size", slice_size_a), in UnrollBatchMatMul3D() 66 CreateInt32Array(model, batch_name + "/slice_a/reshape/shape", in UnrollBatchMatMul3D() 85 CreateInt32Array(model, batch_name + "/slice_b/slice/begin", in UnrollBatchMatMul3D() 87 CreateInt32Array(model, batch_name + "/slice_b/slice/size", slice_size_b), in UnrollBatchMatMul3D() 98 CreateInt32Array(model, batch_name + "/slice_b/reshape/shape", in UnrollBatchMatMul3D() 188 const string perm_array_name = CreateInt32Array( in TransposeInput()
|
D | convert_squeeze_to_reshape.cc | 71 CreateInt32Array(model, squeeze_op->outputs[0] + "_shape", in Run()
|
D | resolve_tensorflow_matmul.cc | 102 CreateInt32Array( in Run()
|
/external/tensorflow/tensorflow/lite/toco/ |
D | tooling_util.h | 268 string CreateInt32Array(Model* model, const string& param_name,
|
D | tooling_util.cc | 1366 CreateInt32Array( in InsertCopyOperator() 1858 string CreateInt32Array(Model* model, const string& param_name, in CreateInt32Array() function
|