Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 107 for Dtype (0.04 sec)

  1. tensorflow/cc/ops/while_loop.cc

      NodeBuilder::NodeOut enter_input(enter_output.node(), enter_output.index());
    
      const int next_output_index = 0;
      DataType dtype = enter_output.node()->output_type(0);
      NodeBuilder::NodeOut next_input(NextIterationName(scope, loop_var_idx),
                                      next_output_index, dtype);
    
      std::vector<NodeBuilder::NodeOut> input_list({enter_input, next_input});
      const string unique_name = scope.GetUniqueNameForOp("Merge");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 01:01:21 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

        size_t idx = type_and_idx.index();
        auto result_ty = mlir::cast<mlir::RankedTensorType>(type_and_idx.value());
    
        // If the result type isn't static, then the owner of the result may be a
        // cast op from a more specific bounded type to an unbounded dynamic type.
        // Use the bounded type to get the buffer size.
        mlir::RankedTensorType buffer_ty = result_ty;
        if (!buffer_ty.hasStaticShape()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_graph.cc

            operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str()));
        TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype));
        if (!shape.unknown_rank()) {
          TF_RETURN_IF_ERROR(operation->SetAttrShape(
              "shape", reinterpret_cast<int64_t*>(shape.dim_sizes().data()),
              shape.dims()));
        }
        int num_outputs = 1;
        std::vector<AbstractTensorHandle*> outputs(num_outputs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 20:00:09 UTC 2024
    - 15.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc

    }
    
    TEST(Dump, TextualIrWithOptions) {
      Graph graph(OpRegistry::Global());
      Node* node;
      TF_ASSERT_OK(NodeBuilder("A", "Placeholder")
                       .Attr("dtype", DT_FLOAT)
                       .Finalize(&graph, &node));
    
      string actual;
      StringWritableFile file(&actual);
      TF_ASSERT_OK(DumpTextualIRToFile(MlirDumpConfig().emit_location_information(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 03:47:51 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/aot/aot_only_var_handle_op.cc

        .Attr("shared_name: string = ''")
        .Attr("debug_name: string = ''")
        .Attr("dtype: type")
        .Attr("shape: shape")
        .Output("resource: resource")
        .SetIsStateful()
        .SetShapeFn([](shape_inference::InferenceContext* c) {
          c->set_output(0, c->Scalar());
          DataType t;
          TF_RETURN_IF_ERROR(c->GetAttr("dtype", &t));
          PartialTensorShape p;
          TF_RETURN_IF_ERROR(c->GetAttr("shape", &p));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 28 09:57:04 UTC 2024
    - 3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

              dtype == tensorflow::DT_UINT16 || dtype == tensorflow::DT_INT16 ||
              dtype == tensorflow::DT_UINT8 || dtype == tensorflow::DT_INT8 ||
              dtype == tensorflow::DT_HALF || dtype == tensorflow::DT_BFLOAT16 ||
              dtype == tensorflow::DT_FLOAT || dtype == tensorflow::DT_DOUBLE ||
              dtype == tensorflow::DT_COMPLEX64 ||
              dtype == tensorflow::DT_COMPLEX128 || dtype == tensorflow::DT_BOOL) {
            return {};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

        for (auto type : inst.getResultTypes())
          if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType,
                         mlir::tf_executor::TokenType>(type))
            return errors::InvalidArgument(
                "Values must be of tensor type, TensorFlow control type, or "
                "TensorFlow token type. Found ",
                mlir::debugString(type));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

        for (auto type : inst.getResultTypes())
          if (!mlir::isa<mlir::TensorType, mlir::tf_executor::ControlType,
                         mlir::tf_executor::TokenType>(type))
            return errors::InvalidArgument(
                "Values must be of tensor type, TensorFlow control type, or "
                "TensorFlow token type. Found ",
                mlir::debugString(type));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc

          // Return values
          {"y: float"},
          // Attr def
          {},
          // Nodes
          {
              {{"two"},
               "Const",
               {},
               {{"value", test::AsScalar<int32>(2)}, {"dtype", DT_INT64}}},
              {{"scale"},
               "Cast",
               {"two"},
               {{"SrcT", DT_INT64}, {"DstT", DT_FLOAT}}},
              {{"y"}, "Mul", {"x", "scale"}, {{"T", DT_FLOAT}}},
          });
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 19:51:50 UTC 2024
    - 10.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/abstract_tensor_handle.cc

        shape_string = "<error computing shape>";
      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 1.4K bytes
    - Viewed (0)
Back to top