Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 47 of 47 for add_input (0.2 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        TFL_StatefulTensor:$fw_input_cell_state,
        TFL_StatefulTensor:$bw_input_activation_state,
        TFL_StatefulTensor:$bw_input_cell_state,
    
        // Auxiliary input & weights.
        TFL_TensorOfOrNone<[F32, I8]>:$aux_input,
        // Auxiliary fw weights.
        TFL_TensorOfOrNone<[F32, I8]>:$fw_aux_input_to_input_weights,
        TFL_TensorOfOrNone<[F32, I8]>:$fw_aux_input_to_forget_weights,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

                 llvm::enumerate(block.getArgumentTypes())) {
              TensorType updated_ty =
                  UpdateElementTypeTo(original_ty, element_types[block_idx]);
              signature.addInputs(block_idx, {updated_ty});
            }
            rewriter.applySignatureConversion(&region.front(), signature);
          }
        }
    
        // Replace all uses of `op` results with the newly created op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top