Searched refs:forward_input_to_output_with_shape (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | guarantee_const_op.cc | 34 if (!ctx->forward_input_to_output_with_shape(0, 0, input_tensor.shape(), in Compute()
|
| H A D | aggregate_ops.cc | 59 if (ctx->forward_input_to_output_with_shape(input_idx, 0, input0.shape(), in Compute()
|
| H A D | scatter_nd_op.cc | 296 if (!c->forward_input_to_output_with_shape(0, 0, params_shape, in DoCompute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 478 bool OpKernelContext::forward_input_to_output_with_shape( in forward_input_to_output_with_shape() function in tensorflow::OpKernelContext
497 Status OpKernelContext::forward_input_to_output_with_shape( in forward_input_to_output_with_shape() function in tensorflow::OpKernelContext
503 if (!forward_input_to_output_with_shape(input_index, output_index, in forward_input_to_output_with_shape()
599 if (forward_input_to_output_with_shape(input_index, output_index, in forward_input_or_allocate_output()
617 if (forward_input_to_output_with_shape(input_name, output_name, in forward_input_or_allocate_output()
|
| H A D | op_kernel.h | 836 bool forward_input_to_output_with_shape(int input_index, int output_index,
839 Status forward_input_to_output_with_shape(StringPiece input_name,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tfrt_op_kernel.h | 134 bool forward_input_to_output_with_shape(int input_index, int output_index, in forward_input_to_output_with_shape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | linalg_ops_common.cc | 210 if (context->forward_input_to_output_with_shape( in PrepareOutputs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_matmul_op_fused.cc | 162 if (native_format && ctx->forward_input_to_output_with_shape( in Compute()
|
| H A D | mkl_conv_ops.cc | 1118 if (native_format && context->forward_input_to_output_with_shape( in AllocateOutputTensor()
2041 ? context->forward_input_to_output_with_shape( in AllocateOutputTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 958 is_forwarded = context->forward_input_to_output_with_shape(
|