Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for input_size (0.12 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

            int64_t output_size;
            int64_t pad_low_int64;
            int64_t pad_high_int64;
            int64_t input_size = input_ty.getDimSize(dim);
            if (input_size == ShapedType::kDynamic) return failure();
            absl::Status status = tensorflow::GetWindowedOutputSizeVerbose(
                input_size, filter_ty.getDimSize(i), dilation, stride, padding,
                &output_size, &pad_low_int64, &pad_high_int64);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        Location loc = op.getLoc();
        Value scalar_zero = CreateI32SplatConst(loc, &rewriter, {}, 0);
    
        // Compute the input tensorlist's length and store it in `input_size`.
        IntegerType shape_dtype = rewriter.getIntegerType(32);
        auto input_size = rewriter.create<TF::TensorListLengthOp>(
            loc, tensorflow::GetTypeFromTFTensorShape({}, shape_dtype),
            op.getOperand(0));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      }];
    
      let arguments = (ins
        Arg<TF_Int32Tensor, [{An integer vector representing the shape of `input`,
    where `input` is a 4-D `[batch, height, width, channels]` tensor.}]>:$input_sizes,
        Arg<TensorOf<[TF_Bfloat16, TF_Float16, TF_Float32, TF_Float64, TF_Int32]>, [{4-D with shape
    `[filter_height, filter_width, in_channels, out_channels]`.}]>:$filter,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top