- Sort Score
- Result 10 results
- Languages All
Results 11 - 19 of 19 for init_values (0.27 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
public: using ConvertReduceOpToTfOp::ConvertReduceOpToTfOp; LogicalResult MatchInitValue(Value init_value) const override { auto type = mlir::cast<ShapedType>(init_value.getType()).getElementType(); if (mlir::isa<FloatType>(type)) { float const_value; if (failed(GetConstantSplatValue<float>(init_value, const_value)) || const_value != 1.0) return failure();
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/quantization/tensorflow/passes/insert_main_function.cc
} else { llvm::SmallVector<Value> input_values(node_output_tensors.size(), scalar_one); for (const auto& [output_index, tensor_index, tensor_value] : node_output_tensors) { input_values[tensor_index] = tensor_value; } identity_op = builder.create<TF::IdentityNOp>( new_loc, TypeRange(ValueRange(input_values)), input_values); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
Type current_input_type = mlir::cast<TypeAttr>(input_types[i]).getValue(); if (current_input_type != target_input_type) { input_values[i] = rewriter.create<TF::CastOp>(loc, result_type, input_values[i]); } } } const SymbolTable& symbol_table_; const bool materialize_derived_attrs_; const llvm::SmallDenseSet<StringRef, 4> fixed_elt_type_attrs_{"i32_", "i64_",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/pipelines/process_nchw_tensor.mlir
// CHECK: %[[ADD:.+]] = stablehlo.add %[[TRANSPOSE_1]], %[[ARG_1]] : tensor<1x4x5x5xf32> // CHECK: return %[[ADD]] // ----- // Tests that a `reduce_window{max}(add(convolution(%activation, %weight), %bias), %init_value)` // with the activation tensor of NCHW format is converted to NHWC convolution + // add + reduce_window (with max) operation. Transpose ops are inserted to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 20:32:46 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
TF_Output output{dequeue_op, 0}; TF_Tensor* ret; TF_SessionRun(session, /*run_options*/ nullptr, // input related parameters /*inputs*/ nullptr, /*input_values*/ nullptr, /*ninputs*/ 0, // output related parameters /*outputs*/ &output, /*output_values*/ &ret, /*noutputs*/ 1, /*targets*/ nullptr, /*ntargets*/ 0,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// workaround. SmallVector<int32_t, 4> new_element_shape_values; auto int_values = dense_elem_attr.getValues<APInt>(); for (auto it = int_values.begin(); it != int_values.end(); ++it) { auto dim_value = (*it).getSExtValue(); if (it == int_values.begin() && dim_value == -1) { if (!default_to_single_batch_) { const char *error_info =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} session->graph->mu.unlock(); status->status = session->session->Extend(std::move(graph_def)); if (!status->status.ok()) { // Contract is we always delete input_values[i]. return false; } // Note: session->session is not modified if Extend() fails, so // we only set last_num_graph_nodes if it succeeds. session->last_num_graph_nodes = num_nodes;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
} OpBuilder builder(attr.getContext()); const auto dense_int_attr = mlir::dyn_cast<DenseIntElementsAttr>(attr); const auto int_values = dense_int_attr.getValues<APInt>(); float float_val = 0.0f; if (!int_values.empty()) { const APInt apint_val = *int_values.begin(); if (dense_int_attr.getType().getElementType().isSignedInteger()) { // Get the sign-extended value (=>int64) if the type is signed.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)