Home
last modified time | relevance | path

Searched refs:GetTensorMetaDataIndex (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util.h878 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
881 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
889 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
892 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
969 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
985 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
1007 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
1028 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
1145 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
1147 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h68 int inline GetTensorMetaDataIndex(int n, int total_tensors) { in GetTensorMetaDataIndex() function
Dmkl_layout_pass.cc1691 GetTensorMetaDataIndex(n_output_slot, n->num_outputs()); in GetNodeProducingMklTensor()
3490 GetTensorMetaDataIndex(n_data_op_slot, n_data->num_outputs()); in FixMklMetaDataEdgeIfNeeded()
3550 GetTensorMetaDataIndex(e->dst_input(), n->num_inputs()); in FixMklMetaDataEdges()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_concat_op.cc480 GetTensorMetaDataIndex(0, context->num_outputs()), in CallEigenVersion()