Home
last modified time | relevance | path

Searched refs:SharesBufferWith (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dunique_tensor_references_test.cc28 EXPECT_FALSE(a.SharesBufferWith(b)); in TEST()
35 if (tensors[0].SharesBufferWith(a)) { in TEST()
36 EXPECT_TRUE(tensors[1].SharesBufferWith(b)); in TEST()
38 EXPECT_TRUE(tensors[1].SharesBufferWith(a)); in TEST()
39 EXPECT_TRUE(tensors[0].SharesBufferWith(b)); in TEST()
51 EXPECT_TRUE(a.SharesBufferWith(b)); in TEST()
58 EXPECT_TRUE(tensors[0].SharesBufferWith(a)); in TEST()
59 EXPECT_TRUE(tensors[0].SharesBufferWith(b)); in TEST()
70 EXPECT_FALSE(a.SharesBufferWith(b)); in TEST()
84 EXPECT_FALSE(a.SharesBufferWith(b)); in TEST()
[all …]
Dtensor_reference.h52 bool SharesBufferWith(const TensorReference& t) const { in SharesBufferWith() function
57 bool SharesBufferWith(const Tensor& t) const { in SharesBufferWith() function
Dunique_tensor_references.h57 return t1.SharesBufferWith(t2); in operator()
Dunique_tensor_references.cc47 if (referenced_tensors_vector_[i].SharesBufferWith(tensor)) { in Add()
Dtensor_test.cc1128 TEST(Tensor, SharesBufferWith) { in TEST() argument
1134 EXPECT_FALSE(a_empty.SharesBufferWith(a_empty)); in TEST()
1135 EXPECT_FALSE(a_empty.SharesBufferWith(b_empty)); in TEST()
1136 EXPECT_FALSE(a_empty.SharesBufferWith(a)); in TEST()
1137 EXPECT_FALSE(a_empty.SharesBufferWith(copy)); in TEST()
1138 EXPECT_TRUE(a.SharesBufferWith(a)); in TEST()
1139 EXPECT_FALSE(a.SharesBufferWith(b)); in TEST()
1140 EXPECT_TRUE(a.SharesBufferWith(copy)); in TEST()
Dtensor.h220 bool SharesBufferWith(const Tensor& b) const;
Dtensor.cc1181 bool Tensor::SharesBufferWith(const Tensor& b) const { in SharesBufferWith() function in tensorflow::Tensor
/external/tensorflow/tensorflow/core/kernels/
Dsnapshot_op.cc38 if (!output->SharesBufferWith(input)) { in Compute()
Ddeterminant_op.cc178 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
323 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
Dself_adjoint_eig_v2_op_gpu.cc108 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
Dlu_op_gpu.cu.cc127 if (!packed_triangular_factors->SharesBufferWith(input)) { in ComputeAsync()
Dstack.cc88 return !tensor.SharesBufferWith(first); in IsUsefulToSwap()
Dmatrix_solve_op.cc193 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
Dsvd_op_gpu.cu.cc281 if (!M.SharesBufferWith(input_copy)) { in PerformSVD_MlessN()
/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc192 if (y_data.SharesBufferWith((*x_datas)[x_idx])) { in EvaluateGraph()