- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 32 for add_input (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
// Rewrite pad_op before Convolutioin. if (failed(HandlePad(pad_op, filter_shape[0], block_size))) return; auto pad_input = pad_op.getInput(); if (auto block_arg = mlir::dyn_cast<mlir::BlockArgument>(pad_input)) { // Change on device function type/shape. HandleFuncOp(block_arg.getOwner()->getParentOp()); } } // Handle Conv2D input, stride and filter.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/prepare_lifting.mlir
// CHECK: %[[q_input:.*]] = "quantfork.qcast"(%arg0) : (tensor<1x3x4x3xf32>) -> tensor<1x3x4x3x!quant.uniform<i8:f32, 0.0011764706057660721:-43>> // CHECK: %[[dq_input:.*]] = "quantfork.dcast"(%[[q_input]]) : (tensor<1x3x4x3x!quant.uniform<i8:f32, 0.0011764706057660721:-43>>) -> tensor<1x3x4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 03:24:59 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
// [0, 1] is symmetric across the line Y=0.5. Value abs_input = rewriter.create<AbsOp>(loc, input); Value abs_input_floor = rewriter.create<FloorOp>(loc, abs_input); Value abs_frac_input = rewriter.create<SubOp>(loc, abs_input, abs_input_floor); // Convert values of abs_frac_input > 0.5 to (1 - frac_input) to improve
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/const-fold.mlir
func.return %5, %6, %7, %8, %9 : tensor<f32>, tensor<4xf32>, tensor<4xf32>, tensor<4xf32>, tensor<4xf32> } // CHECK-LABEL: @add_int func.func @add_int() -> (tensor<i32>, tensor<4xi32>, tensor<4xi32>, tensor<4xi32>) { %0 = arith.constant dense<8> : tensor<i32> %1 = arith.constant dense<1> : tensor<i32> %2 = arith.constant dense< 4> : tensor<4xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 45.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (n->IsOp()) { NodeDef* node_def = n->mutable_def(); node_def->clear_input(); for (const Edge* e : n->in_edges()) { Node* src = e->src(); if (src->IsOp()) { Graph::AddInput(node_def, src->name(), e->src_output()); } } } } } absl::StatusOr<std::unique_ptr<Graph>> Exporter::Convert( const GraphExportConfig& configs, const Dialect* tf_dialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (n->IsOp()) { NodeDef* node_def = n->mutable_def(); node_def->clear_input(); for (const Edge* e : n->in_edges()) { Node* src = e->src(); if (src->IsOp()) { Graph::AddInput(node_def, src->name(), e->src_output()); } } } } } absl::StatusOr<std::unique_ptr<Graph>> Exporter::Convert( const GraphExportConfig& configs, const Dialect* tf_dialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
Region &entry = func.getRegion(); TypeConverter::SignatureConversion signature_conversion( entry.getNumArguments()); for (const BlockArgument &arg : entry.getArguments()) { signature_conversion.addInputs(arg.getArgNumber(), updated_argument_types[arg.getArgNumber()]); } rewriter.applySignatureConversion(&entry.front(), signature_conversion); }
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/eager/c_api.cc
return tensorflow::unwrap(op)->DeviceName().c_str(); } void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input)); } void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInputList(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
::flatbuffers::uoffset_t start_; void add_opcode_index(uint32_t opcode_index) { fbb_.AddElement<uint32_t>(Operator::VT_OPCODE_INDEX, opcode_index, 0); } void add_inputs(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> inputs) { fbb_.AddOffset(Operator::VT_INPUTS, inputs); } void add_outputs(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> outputs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
auto signature_key_buf = builder_.CreateString(signature_def_data.signature_key); tflite::SignatureDefBuilder sig_def_builder(builder_); sig_def_builder.add_inputs(inputs_buf); sig_def_builder.add_outputs(outputs_buf); sig_def_builder.add_signature_key(signature_key_buf); sig_def_builder.add_subgraph_index(signature_def_data.subgraph_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)