Searched defs:pack_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_nearest_upsample.cc | 153 std::vector<int> pack_axis; in Run() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3167 int pack_axis; in TrySimplify() local
3186 PartialTensorShape* pack_output_shape, int* pack_axis, in CheckInputs()
3214 int pack_axis, int64* slice_start_value, bool* found, in GetSliceAxis()
3229 int pack_axis, int64* slice_start_value, in GetSimpleSliceAxis()
3320 int pack_axis, int64* slice_start_value, in GetStridedSliceAxis()
3449 int64 slice_start_value, int pack_axis, in RewriteGraph()
|