Searched refs:y_tensor (Results 1 – 4 of 4) sorted by relevance
77 def __init__(self, y_tensor=None): argument92 self._y_tensor = y_tensor95 if y_tensor:96 self._graph = y_tensor.graph101 def y_tensor(self): member in GradientsDebugger
38 y_tensor = ops.convert_to_tensor(y, dtype=dtype)41 x_tensor, y_tensor, out_idx=index_dtype)
47 y_tensor = ops.convert_to_tensor(y, dtype=dtype)48 out_tensor, idx_tensor = diff_func(x_tensor, y_tensor,
1836 auto y_tensor = y_type.cast<TensorType>(); in inferReturnTypes() local1843 if (x_tensor.getElementType() != y_tensor.getElementType() || in inferReturnTypes()1857 } else if (!y_tensor.hasRank()) { in inferReturnTypes()