Searched defs:forward_input (Results 1 – 7 of 7) sorted by relevance
154 void Broadcast(bool forward_input) { in Broadcast()171 bool forward_input) { in RunTest()233 void DoBroadcast(bool forward_input) { in DoBroadcast()
205 std::vector<int> forward_input; in InitializeNode() local
109 forward_input, loop_vars): argument166 def _create_grad_indexed_slices_init(grad_output_slices, forward_input): argument
757 Value* forward_input = nullptr; in call() local
511 std::unique_ptr<Tensor> OpKernelContext::forward_input( in forward_input() function in tensorflow::OpKernelContext
604 void ExecuteAdd(bool async, bool forward_input, bool tfrt) { in ExecuteAdd()
165 def __init__(self, constructor_input, forward_input=None, desc='', reference_fn=None): argument