Searched refs:input_dim_size (Results 1 – 8 of 8) sorted by relevance
100 int input_dim_size, int offset) { in GetInputDimension() argument106 if (padded_dimension >= input_dim_size) { in GetInputDimension()107 padded_dimension -= input_dim_size; in GetInputDimension()108 const int original_ind = input_dim_size - (1 + offset); in GetInputDimension()
986 const int64 input_dim_size = in HandleReshape() local990 VLOG(2) << "input_dim_size: " << input_dim_size in HandleReshape()993 if (input_dim_size == output_dim_size) { in HandleReshape()999 if (input_dim_size > output_dim_size) { in HandleReshape()1000 TF_RET_CHECK(input_dim_size % output_dim_size == 0) in HandleReshape()1002 const int64 divisor = input_dim_size / output_dim_size; in HandleReshape()1016 if (input_dim_size < output_dim_size) { in HandleReshape()
2558 const int64 input_dim_size = operand_shape.dimensions(dim); in InferDynamicSliceShape() local2564 if (slice_dim_size > input_dim_size) { in InferDynamicSliceShape()2567 slice_dim_size, input_dim_size); in InferDynamicSliceShape()2675 const int64 input_dim_size = operand_shape.dimensions(dim); in InferDynamicUpdateSliceShape() local2682 if (update_dim_size > input_dim_size) { in InferDynamicUpdateSliceShape()2685 update_dim_size, input_dim_size); in InferDynamicUpdateSliceShape()
74 kernel_spatial_dim_size, input_dim_size; member2604 const int64 input_dim_size = in GetConvolutionDetails() local2626 input_dim_size + (base_dilation_factor > 1 ? 0 : inherent_low_padding) + in GetConvolutionDetails()2649 input_dim_size}; in GetConvolutionDetails()2736 c.halo_size, c.input_dim_size)); in PerformSpaceToBatchOnConvolution()
1439 int64_t input_dim_size = in Verify() local1441 if (input_dim_size == ShapedType::kDynamicSize) return success(); in Verify()1443 if (input_dim_size % op.getNumResults() != 0) in Verify()1469 int64_t input_dim_size = in Verify() local1471 if (input_dim_size == ShapedType::kDynamicSize) return success(); in Verify()1500 if (!dynamic_dim_index && total_dim_size != input_dim_size) in Verify()1504 << total_dim_size << " vs " << input_dim_size; in Verify()1506 if (dynamic_dim_index && total_dim_size > input_dim_size) in Verify()1510 << total_dim_size << " vs " << input_dim_size; in Verify()
2043 int64 input_dim_size = operand.base_shape().dimensions(input_sharded_dim); in HandleReshape() local2048 if (input_dim_size % output_dim_size == 0) { in HandleReshape()2050 int64 split_factor = input_dim_size / output_dim_size; in HandleReshape()2065 input_dim_size); in HandleReshape()2088 } else if (output_dim_size % input_dim_size == 0) { in HandleReshape()2090 int64 merge_factor = output_dim_size / input_dim_size; in HandleReshape()
2926 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local2929 if (TensorType::isDynamic(input_dim_size)) return failure(); in matchAndRewrite()2932 int64_t slice_size = input_dim_size / num_splits; in matchAndRewrite()3041 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local3042 if (TensorType::isDynamic(input_dim_size)) return failure(); in matchAndRewrite()3044 assert(((dynamic_dim_index && total_dim_size <= input_dim_size) || in matchAndRewrite()3045 (!dynamic_dim_index && total_dim_size == input_dim_size)) && in matchAndRewrite()3050 split_sizes[*dynamic_dim_index] = input_dim_size - total_dim_size; in matchAndRewrite()
1216 output_dim_size = input_dim_size - kernel_dim_size + 1 (requires: input_dim_size >= kernel_dim_size…