Searched refs:slice_start_value (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3072 int slice_start_value; in TrySimplify() local
3075 &slice_start_value, &found)); in TrySimplify()
3078 return RewriteGraph(node, pack, slice_start_value, pack_axis, in TrySimplify()
3136 int pack_axis, int* slice_start_value, bool* found) { in GetSliceAxis() argument
3236 *slice_start_value = (begin_index == -1) ? 0 : begin_value; in GetSliceAxis()
3239 if (slice_end_value != *slice_start_value + 1) { in GetSliceAxis()
3244 if (*slice_start_value < 0 || *slice_start_value >= pack->input_size()) { in GetSliceAxis()
3247 *slice_start_value, " on axis ", slice_axis, in GetSliceAxis()
3256 int slice_start_value, int pack_axis, in RewriteGraph() argument
3261 GetInputNode(pack->input(slice_start_value), &input_slice)); in RewriteGraph()
[all …]
|