- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for input_tensor (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
] ) def __call__(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]: """Performs a matrix multiplication. Args: input_tensor: Input tensor to matmul with the filter. Returns: A map of: output key -> output result. """ out = math_ops.matmul(input_tensor, self.filters) return {'output': out}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
if (input_tensor.isSplat()) { return input_tensor.reshape(input_tensor.getType().cloneWith( output_shape, input_tensor.getElementType())); } // MLIR implementation pads elements < 8 bits to 8 bits and pads non byte // aligned to the nearest byte. So this is allowed. const char* raw_input = input_tensor.getRawData().data(); const int element_byte_size =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// "third_party/tensorflow/compiler/xla/xla_data.pb.h" into // "third_party/tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc" is // resolved LogicalResult PrecheckForXlaConvV2Op(XlaConvV2Op op) { auto input_tensor = op.getLhs(); auto kernel_tensor = op.getRhs(); auto window_strides = op.getWindowStrides(); auto padding = op.getPadding(); auto lhs_dilation = op.getLhsDilation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
QuantizableResult, Pure]> { let summary = "Mean operator"; let description = [{ Computes the mean of elements across dimensions of a tensor. Reduces input_tensor along the dimensions given in axis. Unless keepdims is true, the rank of the tensor is reduced by 1 for each entry in axis. If keepdims is true, the reduced dimensions are retained with length 1. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)