Searched defs:expected_dtype (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/cc/ops/
H A Dconst_op_test.cc38 void ExpectTypeAndShape(const Node* n, DataType expected_dtype, argument
45 EXPECT_EQ(dtype, expected_dtype);
/external/tensorflow/tensorflow/core/framework/
H A Dtensor.cc612 void Tensor::CheckType(DataType expected_dtype) const {
613 CHECK_EQ(dtype(), expected_dtype); local
616 void Tensor::CheckTypeAndIsAligned(DataType expected_dtype) const {
617 CHECK_EQ(dtype(), expected_dtype); local

Completed in 295 milliseconds