Searched refs:GetTensorDataIndex (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/graph/ |
D | mkl_graph_util.h | 59 int inline GetTensorDataIndex(int n, int total_tensors) { in GetTensorDataIndex() function 71 int tidx = GetTensorDataIndex(n, total_tensors); in GetTensorMetaDataIndex()
|
D | mkl_layout_pass.cc | 3219 GetTensorDataIndex(e->src_output(), e->src()->num_outputs()), in RewriteNode()
|
/external/tensorflow/tensorflow/core/util/ |
D | mkl_util.h | 901 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 …]
|