Searched refs:BuildOnesLike (Results 1 – 3 of 3) sorted by relevance
106 virtual Status BuildOnesLike(const TapeTensor& t,620 TF_RETURN_IF_ERROR(vspace.BuildOnesLike( in InitialGradients()639 vspace.BuildOnesLike(source_tensor->second, &ones_like)); in InitialGradients()959 TF_RETURN_IF_ERROR(vspace_.BuildOnesLike(output_tensor, &aid)); in ForwardpropFromTape()
117 Status BuildOnesLike(const TapeTensor& t,189 Status TapeVSpace::BuildOnesLike(const TapeTensor& t, in BuildOnesLike() function in tensorflow::gradients::TapeVSpace
1300 Status BuildOnesLike(const PyTapeTensor& t, in BuildOnesLike() function in PyVSpace