- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 171 for res_value (0.14 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
bool prefer_tf2xla) { legalize_chlo_ = legalize_chlo; prefer_tf2xla_ = prefer_tf2xla; use_tf2xla_fallback_ = tf2xla_fallback_device_type.has_value(); if (tf2xla_fallback_device_type.has_value()) { device_type_ = tf2xla_fallback_device_type.value().str(); } } /// Performs the lowering to XLA dialect. void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_to_vhlo.cc
auto type = result.getType(); result.setType(converter.convertType(result.getType())); auto new_value = converter.materializeArgumentConversion( rewriter, result.getLoc(), type, {result}); rewriter.replaceAllUsesExcept(result, new_value, new_value.getDefiningOp()); } // Wrap operands in an an unrealized cast to create a cast to buffer any type
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 19:48:51 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/utils.h
auto reducer = builder->create<Op>(loc, block->getArgument(0), block->getArgument(1)); builder->create<ReturnOp>(loc, reducer.getResult()); } ConstantOp GetScalarConstOfType(Type ty, Location loc, int64_t raw_value, OpBuilder* builder); ConstantOp GetScalarNegZeroOfType(Type ty, Location loc, OpBuilder* builder); // Converts an ArrayAttr to a 1D 64-bit dense elements attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
auto lhs_map_or = EquationToMap(lhs); if (!lhs_map_or.has_value()) return std::nullopt; auto lhs_map = lhs_map_or.value(); auto rhs_map_or = EquationToMap(rhs); if (!rhs_map_or.has_value()) return std::nullopt; auto rhs_map = rhs_map_or.value(); auto out_map_or = EquationToMap(out); if (!out_map_or.has_value()) return std::nullopt; auto out_map = out_map_or.value();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
mlir::Value new_value = input.getOutput(); if (mlir::isa<TF::TPUAnnotateTensorsWithDynamicShapeOp>( result.getDefiningOp())) { TF::TPUAnnotateTensorsWithDynamicShapeOp annotate_op = builder.create<TF::TPUAnnotateTensorsWithDynamicShapeOp>( op->getLoc(), result.getType(), new_value, result.getDefiningOp()->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
}; auto op_has_device = [&](Operation& op, InferenceDeviceType& device) -> bool { std::optional<InferenceDeviceType> op_device = GetInferenceDeviceTypeForOp(&op); if (!op_device.has_value()) return false; device = op_device.value(); return true; }; auto op_device_is = [&](Operation& op, llvm::StringRef device) -> bool { InferenceDeviceType device_type;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
if (!IsOpWithQuantizableTrait(op) || FindQuantizationUnitFromLoc(op->getLoc()).has_value()) { return failure(); } std::optional<QuantizationUnit> quantization_unit = FindQuantizationUnit(op); if (!quantization_unit.has_value()) return failure(); if (quantization_unit->func_name().empty()) { std::string func_name =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
return false; } // Check if op or its ancestor uses a key in `virtual_devices`. bool AncestorUsesVirtualDevice( const std::optional<DictionaryAttr>& virtual_devices, Operation* op) { if (!virtual_devices.has_value()) return false; if (!op) return false; if (llvm::isa<tf_device::ReplicateOp>(op)) return false; if (DirectUseOfVirtualDevice(*virtual_devices, op)) return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
// Inversely transposes all the attributes to prepare for the new reduce op auto getInvertPermutedAttr = [&](std::optional<ArrayRef<int64_t>> vals) -> DenseI64ArrayAttr { return vals.has_value() ? DenseI64ArrayAttr::get( ctx, InvertPermutationToVector(*vals, transpose_perm)) : nullptr; }; DenseI64ArrayAttr new_window_dimensions =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
if (input_sharding.has_value() && output_sharding.has_value() && input_sharding.value() != output_sharding.value()) { return absl::InvalidArgumentError(absl::StrCat( "arg#", arg_idx, " is aliased to retval#", retval_idx, " but their sharding configurations don't match.")); } else if (input_sharding.has_value() && !output_sharding.has_value()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0)