| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NaiveDilatedConvolution.cpp | 548 auto is_batch = input.dim() == 4; in slow_conv_dilated2d_cpu() local
600 auto is_batch = input.dim() == 5; in slow_conv_dilated3d_cpu() local
652 auto is_batch = input.dim() == 4; in slow_conv_dilated2d_backward_cpu() local
708 auto is_batch = input.dim() == 5; in slow_conv_dilated3d_backward_cpu() local
|
| H A D | NaiveConvolutionTranspose3d.cpp | 257 bool is_batch = false; in slow_conv_transpose3d_out_cpu_template() local
473 bool is_batch = false; in slow_conv_transpose3d_backward_out_cpu_template() local
691 bool is_batch = false; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
|
| H A D | NaiveConvolutionTranspose2d.cpp | 275 bool is_batch = false; in slow_conv_transpose2d_out_cpu_template() local
462 bool is_batch = false; in slow_conv_transpose2d_backward_out_cpu_template() local
|
| H A D | DilatedConvolutionUtils.h | 142 bool is_batch = input.dim() == dim + 2; in slow_conv_dilated_shape_check() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | NaiveDilatedConvolution.cu | 420 auto is_batch = input.dim() == 4; in slow_conv_dilated2d_cuda() local
469 auto is_batch = input.dim() == 4; in slow_conv_dilated2d_backward_cuda() local
526 auto is_batch = input.dim() == 5; in slow_conv_dilated3d_cuda() local
575 auto is_batch = input.dim() == 5; in slow_conv_dilated3d_backward_cuda() local
|
| H A D | NaiveConvolutionTranspose2d.cu | 188 bool is_batch = false; in slow_conv_transpose2d_out_cuda_template() local
385 bool is_batch = false; in slow_conv_transpose2d_backward_out_cuda_template() local
578 bool is_batch = false; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
|
| H A D | NaiveConvolutionTranspose3d.cu | 260 int is_batch = false; in slow_conv_transpose3d_out_cuda_template() local
486 bool is_batch = false; in slow_conv_transpose3d_backward_out_cuda_template() local
710 bool is_batch = false; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_fast_parsing.cc | 2296 bool is_batch, int num_examples, in ParseContextDenseFeatures()
2359 bool is_batch, int num_examples, in ParseContextSparseFeatures()
2421 bool is_batch, int num_examples, in ParseContextRaggedFeatures()
2499 bool is_batch, int num_examples, in ParseSequenceDenseFeatures()
2653 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceSparseFeatures()
2781 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceRaggedFeatures()
2932 bool is_batch) { in FastParseSequenceExample()
|
| /aosp_15_r20/external/intel-media-driver/media_softlet/linux/common/os/ |
| H A D | mos_oca_defs_specific.h | 109 bool is_batch; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | math_grad.cc | 1035 Status MatMulGradHelper(const Scope& scope, const bool is_batch, in MatMulGradHelper()
1062 const bool is_batch, in MatMulGradCommon()
|
| H A D | math_grad_test.cc | 606 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { in TestMatMulGrad()
631 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, in RandMatMulShapes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | matmul_utils.cc | 56 bool is_batch = absl::c_count(batch_dims, dim) != 0; in GetNonContractingDims() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | example_parsing_ops.cc | 466 bool is_batch = TensorShapeUtils::IsVector(serialized->shape()); in Compute() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PaddingKernel.cpp | 32 bool is_batch = input.dim() == ndim + 2; in PaddingParams() local
|
| /aosp_15_r20/external/libcxx/utils/google-benchmark/include/benchmark/ |
| H A D | benchmark.h | 694 bool is_batch) { in KeepRunningInternal()
|
| /aosp_15_r20/external/google-benchmark/include/benchmark/ |
| H A D | benchmark.h | 1037 bool is_batch) { in KeepRunningInternal()
|
| /aosp_15_r20/external/cronet/third_party/google_benchmark/src/include/benchmark/ |
| H A D | benchmark.h | 996 bool is_batch) { in KeepRunningInternal()
|