Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_TensorType (0.29 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_testlib.h

      std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> actual_value(
          TFE_TensorHandleResolve(handle, status.get()), TF_DeleteTensor);
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      ASSERT_EQ(TF_TensorType(actual_value.get()),
                static_cast<TF_DataType>(DataTypeToEnum<value_type>().value));
      EXPECT_EQ(expected_value,
                *static_cast<value_type*>(TF_TensorData(actual_value.get())));
    }
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 09 01:12:35 GMT 2021
    - 6.9K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

      TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes,
                                  &Deallocator, &deallocator_called);
      EXPECT_FALSE(deallocator_called);
      EXPECT_EQ(TF_FLOAT, TF_TensorType(t));
      EXPECT_EQ(2, TF_NumDims(t));
      EXPECT_EQ(dims[0], TF_Dim(t, 0));
      EXPECT_EQ(dims[1], TF_Dim(t, 1));
      EXPECT_EQ(num_bytes, TF_TensorByteSize(t));
      EXPECT_EQ(static_cast<void*>(values), TF_TensorData(t));
    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)
  3. tensorflow/c/c_test_util.cc

      TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name);
      TF_SetAttrTensor(desc, "value", t, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      TF_SetAttrType(desc, "dtype", TF_TensorType(t));
      *op = TF_FinishOperation(desc, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      ASSERT_NE(*op, nullptr);
    }
    
    TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  4. tensorflow/c/c_api_function_test.cc

        // Check results
        for (int i = 0; i < expected_results.size(); ++i) {
          TF_Tensor* out = csession.output_tensor(i);
          ASSERT_TRUE(out != nullptr);
          EXPECT_EQ(TF_INT32, TF_TensorType(out));
          EXPECT_EQ(1, TF_NumDims(out));
          CompareInt32Tensor(expected_results[i], out);
        }
      }
    
      // Run the host graph, which now contains a function and check that
      // outputs are as expected.
    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)
Back to top