Searched refs:end_value_index (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | array_grad.py | 53 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument 107 return grad + [None] if end_value_index <= dim_index else [None] + grad 110 input_values = op.inputs[start_value_index:end_value_index] 212 return (out_grads + [None] if end_value_index <= dim_index else [None] + 222 end_value_index=len(op.inputs), 229 op, grad, start_value_index=0, end_value_index=-1, dim_index=-1)
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 1886 int end_value_index, int dim_index) { in ConcatShapeHelper() argument 1896 for (int i = start_value_index; i < end_value_index; ++i) { in ConcatShapeHelper() 1909 for (int i = start_value_index; i < end_value_index; ++i) { in ConcatShapeHelper() 1937 ShapeHandle input = c->input(end_value_index - 1); in ConcatShapeHelper() 1947 for (int i = end_value_index - 2; i >= start_value_index; --i) { in ConcatShapeHelper()
|