Searched defs:pack_axis (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer.cc | 3751 int pack_axis; in TrySimplify() local
3770 PartialTensorShape* pack_output_shape, int* pack_axis, in CheckInputs()
3798 int pack_axis, int64_t* slice_start_value, bool* found, in GetSliceAxis()
3813 int pack_axis, int64_t* slice_start_value, in GetSimpleSliceAxis()
3904 int pack_axis, int64_t* slice_start_value, in GetStridedSliceAxis()
4033 int64_t slice_start_value, int pack_axis, in RewriteGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | identify_nearest_upsample.cc | 154 std::vector<int> pack_axis; in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | device_transform_patterns.cc | 218 int pack_axis = pack_op.axis(); in matchAndRewrite() local
|