Searched refs:GetTensorMetaDataIndex (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 681 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
684 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
754 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
776 GetTensorMetaDataIndex(n, ctext->num_outputs()),
825 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
827 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut()
881 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut()
883 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in ForwardMklTensorInToOut()
940 GetTensorMetaDataIndex(idx_data_in, context->num_inputs()); in ForwardMklMetaDataInToOut()
942 GetTensorMetaDataIndex(idx_data_out, context->num_outputs()); in ForwardMklMetaDataInToOut()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 74 int inline GetTensorMetaDataIndex(int n, int total_tensors) { in GetTensorMetaDataIndex() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2210 GetTensorMetaDataIndex(n_output_slot, n->num_outputs()); in GetNodeProducingMklTensor()
3933 GetTensorMetaDataIndex(n_data_op_slot, n_data->num_outputs()); in FixMklMetaDataEdgeIfNeeded()
3994 GetTensorMetaDataIndex(e->dst_input(), n->num_inputs()); in FixMklMetaDataEdges()
|