Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 79 for getElementDtype (0.19 sec)

  1. subprojects/core/src/test/groovy/org/gradle/api/internal/collections/ListElementSourceTest.groovy

            final List<T> value
    
            TypedProviderOfList(Class<T> type, List<T> value) {
                this.type = type
                this.value = value
            }
    
            @Override
            Class<T> getElementType() {
                return type
            }
    
            @Override
            protected Value<List<T>> calculateOwnValue(ValueConsumer consumer) {
                return Value.of(value)
            }
    
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Mon Apr 13 05:06:48 UTC 2020
    - 12.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc

          elemType.isInteger(64))
        return true;
    
      // Complex<F<32>> is allowed.
      if (mlir::isa<ComplexType>(elemType) &&
          mlir::cast<ComplexType>(elemType).getElementType().isF32())
        return true;
    
      // QUINT8 and UI8 are allowed.
      if (mlir::isa<TF::Quint8Type>(elemType) ||
          (elemType.isInteger(8) && mlir::cast<IntegerType>(elemType).isUnsigned()))
        return true;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/utils.h

      for (int64_t i = 0; i < permutation_array.size(); ++i) {
        transposed_shape[i] = input_type.getDimSize(permutation_array[i]);
      }
      auto transposed_type =
          RankedTensorType::get(transposed_shape, input_type.getElementType());
      return transposed_type;
    }
    
    // Returns shape of a ranked tensor.
    // Precondition: output_val's is ranked tensor.
    // Returns a truncated shape when `truncate` is set to true.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc

      Type attribute_type = builder.getI64Type();
      return builder.create<TF::StridedSliceOp>(
          loc,
          RankedTensorType::get(
              {}, mlir::cast<ShapedType>(shape_value.getType()).getElementType()),
          /*input=*/shape_value,
          /*begin=*/Create1DConstValue<int32_t>(builder, loc, {dim}),
          /*end=*/Create1DConstValue<int32_t>(builder, loc, {dim + 1}),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  5. subprojects/core/src/main/java/org/gradle/api/internal/collections/AbstractIterationOrderRetainingElementSource.java

            }
            return element;
        }
    
        Element<T> cachingElement(CollectionProviderInternal<T, ? extends Iterable<T>> provider) {
            final Element<T> element = new Element<>(provider.getElementType(), new ElementsFromCollectionProvider<>(provider), this::doAddRealized);
            if (provider instanceof ChangingValue) {
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Mon Mar 04 20:04:06 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc

          // dimension with size 1.
          output_shape[i] = 1;
        } else {
          output_shape[i] = *shape_itr;
          shape_itr++;
        }
      }
    
      return RankedTensorType::get(output_shape, tensor_type.getElementType());
    }
    
    // Determines the output type of the `SliceOp` when it is being inserted in
    // place of a `XlaGatherOp`. When the dimensions of `xla_gather_op_output_type`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 13.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h

      for (int i = 0, e = rank; i != e; i++) {
        int64_t dim0 = a.getDimSize(i);
        int64_t dim1 = b.getDimSize(i);
        dims[i] = (dim0 == ShapedType::kDynamic) ? dim1 : dim0;
      }
      return RankedTensorType::get(dims, a.getElementType());
    }
    }  // namespace detail
    
    // Verifies that op has the same operand and result element types (or type
    // itself, if scalar) after resolving reference types (i.e., after converting
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

    Type ToLegalType(Type type) {
      if (IsTFQintType(type)) return GetIntTypeFromTFQint(type);
      if (auto shaped = mlir::dyn_cast<ShapedType>(type)) {
        Type elem = shaped.getElementType();
        if (IsTFQintType(elem)) return shaped.clone(ToLegalType(elem));
      }
      return type;
    }
    
    bool IsQintToIntCast(Operation *op) {
      auto cast_op = llvm::dyn_cast<TF::CastOp>(op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc

        }
    
        // Only the composite functions with f32 inputs are quantizable.
        if (call_op.getResults().size() == 1 &&
            !mlir::cast<ShapedType>(call_op->getResult(0).getType())
                 .getElementType()
                 .isF32()) {
          check_status.Update(absl::InternalError(
              "Composite functions for quantization should be f32 type."));
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 16.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td

    def CreateTFReadVariableOp : NativeCodeCall<
        "$_builder.create<TF::ReadVariableOp>("
        "  $0.getLoc(),"
        "  GetResourceSubtypeOrDefault("
        "    $2, $1.getType().cast<TensorType>().getElementType()),"
        "  $2)"
        >;
    
    def CheckHasResourceSubtype : Constraint<CPred<"HasResourceSubtype($0)">>;
    
    def CreateConstBoolAttrFalse : NativeCodeCall<"$_builder.getBoolAttr(false)">;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 20.7K bytes
    - Viewed (0)
Back to top