Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsnapshot_op.cc38 if (!output->SharesBufferWith(input)) { in Compute()
Dstack.cc89 return !tensor.SharesBufferWith(first); in IsUsefulToSwap()
Dfused_batch_norm_op.cc932 if (!batch_mean->SharesBufferWith(estimated_mean) && in operator ()()
943 if (!batch_var->SharesBufferWith(estimated_variance) && in operator ()()
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc1161 TEST(Tensor, SharesBufferWith) { in TEST() argument
1167 EXPECT_FALSE(a_empty.SharesBufferWith(a_empty)); in TEST()
1168 EXPECT_FALSE(a_empty.SharesBufferWith(b_empty)); in TEST()
1169 EXPECT_FALSE(a_empty.SharesBufferWith(a)); in TEST()
1170 EXPECT_FALSE(a_empty.SharesBufferWith(copy)); in TEST()
1171 EXPECT_TRUE(a.SharesBufferWith(a)); in TEST()
1172 EXPECT_FALSE(a.SharesBufferWith(b)); in TEST()
1173 EXPECT_TRUE(a.SharesBufferWith(copy)); in TEST()
Dtensor.h282 bool SharesBufferWith(const Tensor& b) const;
Dtensor.cc1245 bool Tensor::SharesBufferWith(const Tensor& b) const { in SharesBufferWith() function in tensorflow::Tensor
/external/tensorflow/tensorflow/core/kernels/linalg/
Dself_adjoint_eig_v2_op_gpu.cc108 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
Ddeterminant_op.cc178 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
323 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
Dlu_op_gpu.cu.cc129 if (!packed_triangular_factors->SharesBufferWith(input)) { in ComputeAsync()
Dmatrix_solve_op.cc193 if (!input.SharesBufferWith(input_copy)) { in ComputeAsync()
Dsvd_op_gpu.cu.cc332 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()