Searched refs:ConstTensorMap (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_input_ops.h | 724 ConstTensorMap;
735 ConstTensorMap A(out_backprop_data + output_offset * image_id,
737 ConstTensorMap B(filter_data, filter_total_size, dims.out_depth);
|
| H A D | conv_grad_ops_3d.cc | 538 ConstTensorMap; in Compute() typedef
549 ConstTensorMap A(out_backprop_data + output_offset * image_id, in Compute()
551 ConstTensorMap B(filter_data, filter_total_size, dims.out_depth); in Compute()
1051 ConstTensorMap; in Compute() typedef
1100 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
1102 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
|
| H A D | conv_grad_filter_ops.cc | 567 ConstTensorMap; in Compute() typedef
604 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
606 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
|