Home
last modified time | relevance | path

Searched refs:GetTensorDataIndex (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h59 int inline GetTensorDataIndex(int n, int total_tensors) { in GetTensorDataIndex() function
71 int tidx = GetTensorDataIndex(n, total_tensors); in GetTensorMetaDataIndex()
Dmkl_layout_pass.cc3219 GetTensorDataIndex(e->src_output(), e->src()->num_outputs()), in RewriteNode()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h901 return ctext->input(GetTensorDataIndex(n, ctext->num_inputs())); in MklGetInput()
1004 ctext, ctext->allocate_output(GetTensorDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
1025 ctext, ctext->allocate_output(GetTensorDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
1144 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
1146 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut()
1167 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyTfTensorInToOutWithShape()
1168 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyTfTensorInToOutWithShape()
1185 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyTfTensorInToOutWithShape()
1186 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyTfTensorInToOutWithShape()
1205 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardTfTensorInToOut()
[all …]