Home
last modified time | relevance | path

Searched refs:total_tensors (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h53 inline int DataIndexToMetaDataIndex(int n, int total_tensors) { in DataIndexToMetaDataIndex() argument
61 return n + total_tensors / 2; in DataIndexToMetaDataIndex()
65 int inline GetTensorDataIndex(int n, int total_tensors) { in GetTensorDataIndex() argument
74 int inline GetTensorMetaDataIndex(int n, int total_tensors) { in GetTensorMetaDataIndex() argument
77 int tidx = GetTensorDataIndex(n, total_tensors); in GetTensorMetaDataIndex()
78 return DataIndexToMetaDataIndex(tidx, total_tensors); in GetTensorMetaDataIndex()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc322 const int total_tensors = param.num_inputs + 1; in TEST_P() local
323 interpreter.AddTensors(total_tensors); in TEST_P()
344 interpreter.SetTensorParametersReadWrite(/*tensor_index=*/total_tensors - 1, in TEST_P()
352 interpreter.SetOutputs({total_tensors - 1}); in TEST_P()
366 /*outputs=*/{total_tensors - 1}, in TEST_P()