Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for cond_input (0.21 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    // WhileOp
    //===----------------------------------------------------------------------===//
    
    static LogicalResult VerifyWhileTypes(Operation *op, TypeRange cond_input,
                                          TypeRange body_input,
                                          TypeRange body_result,
                                          bool shape_invariant) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

        if (!CreateInput(inputs[i], cond_graph, StrCat("cond_input", i).c_str(),
                         &cond_inputs[i], status)) {
          break;
        }
        if (!CreateInput(inputs[i], body_graph, StrCat("body_input", i).c_str(),
                         &body_inputs[i], status)) {
          break;
        }
      }
    
      TF_WhileParams params = {ninputs,    cond_graph,  cond_inputs,  cond_output,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

          }
    
          for (const auto& bound_input :
               llvm::enumerate(concrete_function.bound_inputs())) {
            int arg_index = bound_input_base + bound_input.index();
            auto symbol_ref = mlir::SymbolRefAttr::get(
                builder.getContext(),
                object_names.GetSymbolTableName(bound_input.value()));
            func.setArgAttr(arg_index, "tf_saved_model.bound_input", symbol_ref);
          }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

          ]
      )
      def conv(self, conv_input: core.Tensor) -> Mapping[str, core.Tensor]:
        """Performs a 2D convolution operation.
    
        Args:
          conv_input: Input tensor to perform convolution on.
    
        Returns:
          A map of: output key -> output result.
        """
        out = nn_ops.conv2d(
            conv_input,
            self.conv_filters,
            strides=[1, 1, 2, 1],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

            return rewriter.notifyMatchFailure(
                conv_op, "requires padding to be SAME or VALID");
          }
          padding = "SAME";
        }
        auto conv_input = InsertTranspose(
            conv_op.getLhs(), dnums.getInputBatchDimension(),
            dnums.getInputFeatureDimension(), dnums.getInputSpatialDimensions(),
            /*default_batch_dim=*/0,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

          For example, this would transform
    
          "tf_saved_model.global_tensor"() { sym_name = "v" ... }
          func @f(%arg0 {tf_saved_model.bound_input = @v}) {
            %1 = "tf.ReadVariableOp"(%arg0)
            ...
          }
    
          to
    
          func @f(%arg0 {tf_saved_model.bound_input = @v}) {
            %0 = "tf.VarHandleOp"(sym_name = "v")
            %1 = "tf.ReadVariableOp"(%0)
            ...
          }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
Back to top