Home
last modified time | relevance | path

Searched refs:input_dim_size (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc100 int input_dim_size, int offset) { in GetInputDimension() argument
106 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()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc986 const int64 input_dim_size = in HandleReshape() local
990 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()
Dshape_inference.cc2558 const int64 input_dim_size = operand_shape.dimensions(dim); in InferDynamicSliceShape() local
2564 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() local
2682 if (update_dim_size > input_dim_size) { in InferDynamicUpdateSliceShape()
2685 update_dim_size, input_dim_size); in InferDynamicUpdateSliceShape()
Dspace_to_batch_converter.cc74 kernel_spatial_dim_size, input_dim_size; member
2604 const int64 input_dim_size = in GetConvolutionDetails() local
2626 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()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1439 int64_t input_dim_size = in Verify() local
1441 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() local
1471 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()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2043 int64 input_dim_size = operand.base_shape().dimensions(input_sharded_dim); in HandleReshape() local
2048 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()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2926 int64_t input_dim_size = input_type.getDimSize(dim_index); in matchAndRewrite() local
2929 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() local
3042 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()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DREADME.md1216 output_dim_size = input_dim_size - kernel_dim_size + 1 (requires: input_dim_size >= kernel_dim_size…