Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for scalar (0.17 sec)

  1. tensorflow/c/c_api_test.cc

    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  2. tensorflow/c/eager/unified_api_testutil.h

          TestTensorHandleWithDims<T, datatype>(eager_ctx, data, dims, num_dims);
      *tensor =
          unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get()));
      return absl::OkStatus();
    }
    
    // Return a scalar tensor handle with given value.
    template <class T, TF_DataType datatype>
    Status TestScalarTensorHandle(AbstractContext* ctx, const T value,
                                  AbstractTensorHandle** tensor) {
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/unified_api_test.cc

      }
    
     public:
      bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; }
      bool UseFunction() const { return std::get<2>(GetParam()); }
    };
    
    // Checks that inputs[0] is a scalar.
    Status TestScalarShape(AbstractContext* ctx,
                           absl::Span<AbstractTensorHandle* const> inputs,
                           absl::Span<AbstractTensorHandle*> outputs) {
      PartialTensorShape shape;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 6.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_test_util.h

    #include "tensorflow/core/protobuf/tensorflow_server.pb.h"
    
    // Return a tensor handle containing a float scalar
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value);
    
    // Return a tensor handle containing a int scalar
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, int value);
    
    // Return a tensor handle containing a bool scalar
    TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, bool value);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Jul 17 23:43:59 GMT 2023
    - 7.7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradient_checker.cc

      int num_dims_out = TF_NumDims(model_out_tensor);
      TF_DeleteTensor(model_out_tensor);
    
      // If the output is a scalar, then return the scalar output
      if (num_dims_out == 0) {
        outputs[0] = model_out.release();
        return absl::OkStatus();
      }
    
      // Else, reduce sum the output to get a scalar
    
      // Will sum all dimensions, so get a Tensor containing [0,...,num_dims_out-1].
      AbstractTensorHandlePtr sum_dims;
      {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_function_test.cc

       *                   v
       */
      // Define
      TF_Operation* c = ScalarConst(10, func_graph_, s_, "scalar10");
      Define(-1, {}, {}, {c}, {});
    
      // Use, run, and verify
      TF_Operation* func_op = Use({});
      Run({}, func_op, 10);
      VerifyFDef({"scalar10_0"}, {}, {{"scalar10", DT_INT32}},
                 {{"scalar10_0:output:0", "scalar10"}}, {});
    }
    
    TEST_F(CApiFunctionTest, OneOp_OneInput_OneOutput) {
      /*
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  7. tensorflow/c/c_api_experimental_test.cc

      TFE_OpSetAttrType(fill_op, "T", TF_FLOAT);
      TFE_OpSetAttrType(fill_op, "Tshape", TF_INT32);
    
      float five = 5.0;
      TFE_TensorHandle* scalar = TestScalarTensorHandle(tfe_context_, five);
      TF_Tensor* scalarTensor = TFE_TensorHandleResolve(scalar, status_);
      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
      CheckOutputShapes(fill_op,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jan 17 22:27:52 GMT 2023
    - 13.1K bytes
    - Viewed (1)
  8. tensorflow/c/eager/parallel_device/parallel_device_lib.h

      // Construct a parallel tensor consisting of the scalar values from `values`.
      template <typename DataType>
      std::unique_ptr<ParallelTensor> ScalarsFromSequence(
          absl::Span<const DataType> values, TFE_Context* context,
          TF_Status* status) const;
    
      // A parallel tensor with scalar integers numbering component devices.
      std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  9. tensorflow/c/eager/immediate_execution_context.h

    // Abstract interface to a context.
    //
    // A context is responsible for creating key objects such as Tensors,
    // TensorHandles & Operations.
    class ImmediateExecutionContext : public AbstractContext {
     public:
      // Optimized scalar creation functions
      virtual AbstractTensorInterface* CreateInt64Scalar(int64_t value) = 0;
      virtual AbstractTensorInterface* CreateUint64Scalar(uint64 value) = 0;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 06 08:34:00 GMT 2023
    - 12.3K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_experimental.h

    // Platform-specific implementation to return an unused port. (This should used
    // in tests only.)
    TF_CAPI_EXPORT int TF_PickUnusedPortOrDie(void);
    
    // Fast path method that makes constructing a single scalar tensor require less
    // overhead and copies.
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromScalar(
        TF_DataType data_type, void* data, size_t len, TF_Status* status);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 15.1K bytes
    - Viewed (0)
Back to top