Searched refs:TFE_TensorHandleDataType (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 126 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in MatMulOp()
139 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in ShapeOp()
171 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input)); in MinOp()
|
| D | c_api.h | 163 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
|
| D | c_api_test.cc | 443 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()
1490 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(value_handle)); in TEST()
1533 CHECK_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in BM_ReadVariable()
1608 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()
|
| D | c_api.cc | 440 TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { in TFE_TensorHandleDataType() function
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 305 CHECK_EQ(TFE_TensorHandleDataType(sym_handle), TF_FLOAT); in TEST()
390 CHECK_EQ(TFE_TensorHandleDataType(retvals[0]), TF_FLOAT); in TEST_F()
415 CHECK_EQ(dtype, TFE_TensorHandleDataType(axis)); in TEST_F()
|
| D | c_api_experimental.cc | 8686 TF_DataType inputType = TFE_TensorHandleDataType(tensor); in TFE_EnqueueNamedTensor()
8700 TF_DataType inputType = TFE_TensorHandleDataType(tensor); in TFE_EnqueueNamedTensorFromCtx()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 421 TF_DataType handle_dtype = TFE_TensorHandleDataType(handle.get()); in EagerTensor_init()
444 handle_dtype = TFE_TensorHandleDataType(handle.get()); in EagerTensor_init()
533 return PyIntFromDataType(TFE_TensorHandleDataType(self->handle)); in EagerTensor_datatype_enum()
855 return static_cast<tensorflow::DataType>(TFE_TensorHandleDataType( in PyEagerTensor_Dtype()
|
| D | pywrap_tfe_src.cc | 2185 TF_DataType input_dtype = TFE_TensorHandleDataType(handle->get());
2322 TF_DataType output_dtype = TFE_TensorHandleDataType(handle.get());
2329 output_dtype = TFE_TensorHandleDataType(handle.get());
2382 auto dtype = TFE_TensorHandleDataType(input_handle);
2718 attr_value[j] = TFE_TensorHandleDataType(input_handle);
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor_handle.go | 71 return DataType(C.TFE_TensorHandleDataType(th.c))
|