Searched refs:end_value_index (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 47 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
101 return grad + [None] if end_value_index <= dim_index else [None] + grad
104 input_values = op.inputs[start_value_index:end_value_index]
207 if end_value_index <= dim_index else [None] + out_grads)
216 end_value_index=len(op.inputs),
223 op, grad, start_value_index=0, end_value_index=-1, dim_index=-1)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1212 int end_value_index, int dim_index) { in ConcatShapeHelper() argument
1222 for (int i = start_value_index; i < end_value_index; ++i) { in ConcatShapeHelper()
1235 for (int i = start_value_index; i < end_value_index; ++i) { in ConcatShapeHelper()
1258 ShapeHandle input = c->input(end_value_index - 1); in ConcatShapeHelper()
1268 for (int i = end_value_index - 2; i >= start_value_index; --i) { in ConcatShapeHelper()
|