Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 32 for add_input (0.22 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top