- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 105 for gather (0.27 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
// CHECK: stablehlo.gather // CHECK-NOT: tfl.gather_nd // CHECK-NOT: tfl.gather // ----- // Tests that a float `stablehlo.gather` is not converted to `tfl.gather_nd`. func.func @gather_float(%arg0: tensor<3x4x2x2xf32>, %arg1: tensor<2x3x2xi64>) -> tensor<2x3x2x2xf32> { %0 = "stablehlo.gather"(%arg0, %arg1) { dimension_numbers = #stablehlo.gather< offset_dims = [2, 3],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
return success(); } // Convert gather op to tf.slice and tf.concat LogicalResult ConvertGatherOpToSlice( mhlo::GatherOp gather_op, ConversionPatternRewriter& rewriter) const { Value operand = gather_op.getOperand(); Value start_indices = gather_op.getStartIndices(); static const int rank_two = 2; // This converts a gather op to multiple slice ops, cap the number of slice
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
%0 = "tf.Gather"(%arg0, %arg1) : (tensor<2x3x6xf32>, tensor<4x5xi32>) -> tensor<4x5x3x6xf32> func.return %0 : tensor<4x5x3x6xf32> // CHECK-LABEL:gatherHigherRankIndices // CHECK: "tfl.gather"(%arg0, %arg1) <{axis = 0 : i32, batch_dims = 0 : i32}> : (tensor<2x3x6xf32>, tensor<4x5xi32>) -> tensor<4x5x3x6xf32> } func.func @gatherNdVectorIndices(%arg0 : tensor<3x2x2xf32>, %arg1 : tensor<2xi32>) -> tensor<2xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
}]; } def TFL_GatherOp : TFL_Op<"gather", [ Pure, QuantizableResult, SameOperandsAndResultsScale, TFL_OperandHasAtleastRank<0, 1>, PredOpTrait<"params and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>>, DynamicRangeQuantizedOpInterface ]> { let summary = "Gather operator"; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
func.func @testGather(%arg0 : tensor<?xf32>, %arg1 : tensor<?xi32>) -> tensor<?xf32> { %0 = "tfl.gather"(%arg0, %arg1) {axis = 1 : i32}: (tensor<?xf32>,tensor<?xi32>) -> tensor<?xf32> func.return %0 : tensor<?xf32> } // ----- // CHECK-LABEL: testGather func.func @testGather(%arg0 : tensor<2xf32>, %arg1 : tensor<2xi32>) -> tensor<2xf32> { %0 = "tfl.gather"(%arg0, %arg1) {axis = 1 : i32}: (tensor<2xf32>,tensor<2xi32>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
cmd/metrics-v2.go
// // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as published by // the Free Software Foundation, either version 3 of the License, or // (at your option) any later version. // // This program is distributed in the hope that it will be useful // but WITHOUT ANY WARRANTY; without even the implied warranty of
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Thu Jun 13 22:26:54 UTC 2024 - 131.9K bytes - Viewed (0) -
src/cmd/internal/obj/x86/asm6.go
AVGATHERDPS, AVGATHERQPS, AVPGATHERDD, AVPGATHERQD, AVPGATHERDQ, AVPGATHERQQ: if p.GetFrom3() == nil { // gathers need a 3rd arg. See issue 58822. ctxt.Diag("need a third arg for gather instruction: %v", p) return } // AVX512 gather requires explicit K mask. if p.GetFrom3().Reg >= REG_K0 && p.GetFrom3().Reg <= REG_K7 { if !avx512gatherValid(ctxt, p) { return }
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 15 15:44:14 UTC 2024 - 146.9K bytes - Viewed (0) -
src/cmd/internal/obj/s390x/asmz.go
op_VGFM uint32 = 0xE7B4 // VRR-c VECTOR GALOIS FIELD MULTIPLY SUM op_VGFMA uint32 = 0xE7BC // VRR-d VECTOR GALOIS FIELD MULTIPLY SUM AND ACCUMULATE op_VGEF uint32 = 0xE713 // VRV VECTOR GATHER ELEMENT (32) op_VGEG uint32 = 0xE712 // VRV VECTOR GATHER ELEMENT (64) op_VGBM uint32 = 0xE744 // VRI-a VECTOR GENERATE BYTE MASK op_VGM uint32 = 0xE746 // VRI-b VECTOR GENERATE MASK op_VISTR uint32 = 0xE75C // VRR-a VECTOR ISOLATE STRING
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Apr 16 17:46:09 UTC 2024 - 176.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
constexpr size_t kFbAlignment = 16; // Set `isSigned` to false if the `type` is an 8-bit unsigned integer type. // Since tflite doesn't support unsigned for other types, returns error if // `isSigned` is set to false for other types. static StatusOr<tflite::TensorType> GetTFLiteType(Type type, bool is_signed = true) { if (!is_signed) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} //===----------------------------------------------------------------------===// // Gather op //===----------------------------------------------------------------------===// LogicalResult GatherOp::verify() { GatherOp op = *this; ShapedType params_type = op.getParams().getType().cast<ShapedType>(); // TFLite gather kernel supports 1D string input only. if (params_type.getElementType().isa<mlir::TF::StringType>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)