Searched defs:input_batch (Results 26 – 33 of 33) sorted by relevance
12
1549 auto input_batch = input_tensor_dims[dnums.input_batch_dimension()]; in InferXlaConvOutputShape() local1665 auto input_batch = input_tensor_shape[dnums.input_batch_dimension()]; in PrecheckForXlaConvV2Op() local
811 int64_t input_batch = packed_v_input.sizes()[Layout::BatchMatrices::batch]; in run_baddbmm_context() local
947 int64_t input_batch = in HandleConvolution() local1107 int64_t input_batch = 1; in HandleFft() local
334 int64_t input_batch = input_type.getDimSize(0); in verify() local396 int64_t input_batch = input_shape[0]; in verify() local
1741 const int64_t input_batch = lhs.dimensions(dnums.input_batch_dimension()); in InferConvolveShape() local
402 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4() local
403 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4() local