Searched refs:kOutputIdx (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_aggregate_ops.cc | 89 const size_t kOutputIdx = 0; in ComputeScalar() local
102 output_tf_shape = MklGetInput(ctx, kOutputIdx).shape(); in ComputeScalar()
103 AllocateOutputSetMklShape(ctx, kOutputIdx, &dst_tensor, output_tf_shape, in ComputeScalar()
118 const size_t kOutputIdx = 0; in Compute() local
125 ForwardMklTensorInToOut(ctx, kSrc0Idx, kOutputIdx); in Compute()
127 ForwardTfTensorInToOut(ctx, kSrc0Idx, kOutputIdx); in Compute()
146 AllocateOutputSetMklShape(ctx, kOutputIdx, &dst_tensor, output_tf_shape, in Compute()
233 AllocateOutputSetMklShape(ctx, kOutputIdx, &dst_tensor, output_tf_shape, in Compute()
|
| D | mkl_identity_op.cc | 39 const int kInputIdx = 0, kOutputIdx = 0; in Compute() local
43 ForwardMklTensorInToOut(context, kInputIdx, kOutputIdx); in Compute()
45 ForwardTfTensorInToOut(context, kInputIdx, kOutputIdx); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 354 const int kOutputIdx = 0; in Compute() local
355 AllocateOutputSetMklShape(context, kOutputIdx, &diff_src_tensor, in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 402 const int kOutputIdx = 0; in Compute() local
403 AllocateOutputSetMklShape(context, kOutputIdx, &diff_filter_tensor, in Compute()
|