Searched refs:mkl_shape_output (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_reshape_op.cc | 155 MklDnnShape mkl_shape_output; in Compute() local
156 mkl_shape_output.SetMklTensor(false); in Compute()
159 shape_to, mkl_shape_output); in Compute()
186 MklDnnShape mkl_shape_output; in Compute() local
187 mkl_shape_output.SetMklTensor(true); in Compute()
188 mkl_shape_output.SetMklLayout(&output_tf_pd); in Compute()
189 mkl_shape_output.SetElemType(MklDnnType<T>()); in Compute()
190 mkl_shape_output.SetTfLayout(output_dims.size(), output_dims, in Compute()
196 context, kInputSlotIdx, kOutputSlotIdx, mkl_shape_output); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1171 MklShape mkl_shape_output; in CopyTfTensorInToOutWithShape() local
1172 mkl_shape_output.SetMklTensor(false); in CopyTfTensorInToOutWithShape()
1173 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()
1189 MklDnnShape mkl_shape_output; in CopyTfTensorInToOutWithShape() local
1190 mkl_shape_output.SetMklTensor(false); in CopyTfTensorInToOutWithShape()
1191 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()
1208 MklShape mkl_shape_output; in ForwardTfTensorInToOut() local
1209 mkl_shape_output.SetMklTensor(false); in ForwardTfTensorInToOut()
1210 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in ForwardTfTensorInToOut()
1261 MklDnnShape mkl_shape_output; in SetDummyMklDnnShapeOutput() local
[all …]
|