- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 1,825 for Auto (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
if (auto while_op = llvm::dyn_cast<TF::WhileOp>(use.getOwner())) { auto body = while_op.body_function(); assert(body); auto type_from_body = GetElementTypeFromAccess( body.getArgument(use.getOperandNumber()), module, infer_from_op); if (type_from_body.has_value()) return type_from_body; } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(use.getOwner())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc
std::tuple<StringRef, StringRef, StringRef> GetResourceKey(Operation* op) { StringRef device; if (auto attr = op->getAttrOfType<StringAttr>("device")) { device = attr.getValue(); } StringRef container; if (auto attr = op->getAttrOfType<StringAttr>("container")) { container = attr.getValue(); } StringRef shared_name; if (auto attr = op->getAttrOfType<StringAttr>("shared_name")) { shared_name = attr.getValue(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
OperationState op_state(loc, op_name); for (auto input_num : op.inputs) { if (input_num == -1) { assert(optional_arg_marker != nullptr); op_state.addOperands({optional_arg_marker}); } else { op_state.addOperands({vals_map.at(input_num)}); } } for (auto output_num : op.outputs) { auto& tensor = *tensors.at(output_num); auto type_or_err = tfl::GetTensorType(tensor, builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/jit/device_executable_persistor_test.cc
Scope scope = Scope::NewRootScope().ExitOnError(); auto a = ops::_Arg(scope.WithOpName("A"), DT_INT32, 0); auto b = ops::_Arg(scope.WithOpName("B"), DT_INT32, 1); if (mul) { auto c = ops::Mul(scope.WithOpName("C"), a, b); auto d = ops::_Retval(scope.WithOpName("D"), c, 0); TF_RETURN_IF_ERROR(scope.ToGraph(graph.get())); } else { auto c = ops::Add(scope.WithOpName("C"), a, b);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
std::vector<Output>* grad_outputs) { auto x = op.input(0); auto a = op.input(1); // [Rank(x), 2] // Takes a slice of a. The 1st column. [Rank(x), 1]. auto size = Stack(scope, {Rank(scope, x), 1}); auto pad_before = Slice(scope, a, {0, 0}, size); // Make it a 1-D tensor. auto begin = Reshape(scope, pad_before, {-1});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
auto attr = mlir::dyn_cast<ArrayAttr>(named_attr.getValue()); if (!attr) { return op->emitError() << "'" << kTfSavedModelIndexPathAttr << "' attribute should be an ArrayAttr"; } for (auto element : attr) { if (mlir::isa<StringAttr>(element)) { continue; } if (auto integer = mlir::dyn_cast<IntegerAttr>(element)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
// Create Uniform Quantized constant for the filter. auto subtract_op = cast<stablehlo::SubtractOp>(*op.getResult().user_begin()); auto other_conv_op = cast<stablehlo::ConvolutionOp>( subtract_op.getOperand(1).getDefiningOp()); auto combined_scale_multiply_op = cast<stablehlo::MulOp>(*subtract_op.getResult().user_begin()); auto scale_combined_broadcast_in_dim_op = cast<stablehlo::BroadcastInDimOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
auto symRef = callee.get<SymbolRefAttr>(); return setCalleeAttr(cast<FlatSymbolRefAttr>(symRef)); } // Indirect call, callee Value is the first operand. return setOperand(0, callee.get<Value>()); } LogicalResult ConstantTensorOp::verify() { ConstantTensorOp op = *this; auto input_type = op.getArg().getType(); auto output_type = op.getOut().getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
OpBuilder builder(func); func.walk([&](SparseOpInterface sparse_op) { const auto& sparse_operands = sparse_op.GetSparseOperands(); std::vector<std::vector<int>> supported_block_size; for (int operand : sparse_operands) { auto* op = sparse_op.getOperation(); auto value = op->getOperand(operand); auto* inst = value.getDefiningOp(); if (!inst) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
auto window_strides__ = window_strides ? _fbb.CreateVector<int64_t>(*window_strides) : 0; auto padding__ = padding ? _fbb.CreateVector<int64_t>(*padding) : 0; auto lhs_dilation__ = lhs_dilation ? _fbb.CreateVector<int64_t>(*lhs_dilation) : 0; auto rhs_dilation__ = rhs_dilation ? _fbb.CreateVector<int64_t>(*rhs_dilation) : 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)