Searched refs:pack_axis (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_nearest_upsample.cc | 154 std::vector<int> pack_axis; in Run() local
166 pack_axis.push_back(imaged_original_shape_current_axis + 1); in Run()
179 if (pack_axis.empty()) { in Run()
229 for (int i = pack_axis.size() - 1; i >= 0; --i) { in Run()
236 int axis = pack_axis.at(i); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 218 int pack_axis = pack_op.axis(); in matchAndRewrite() local
220 if (pack_axis < 0) { in matchAndRewrite()
221 pack_axis += rank; in matchAndRewrite()
227 if (i == pack_axis) { in matchAndRewrite()
236 if (i == pack_axis) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3751 int pack_axis; in TrySimplify() local
3753 CheckInputs(node, pack, &pack_output_shape, &pack_axis, &return_early)); in TrySimplify()
3759 TF_RETURN_IF_ERROR(GetSliceAxis(node, pack, pack_output_shape, pack_axis, in TrySimplify()
3764 return RewriteGraph(node, pack, slice_start_value, pack_axis, in TrySimplify()
3770 PartialTensorShape* pack_output_shape, int* pack_axis, in CheckInputs() argument
3775 *pack_axis = pack->attr().at("axis").i(); in CheckInputs()
3784 if (*pack_axis < 0) { in CheckInputs()
3785 *pack_axis += pack_output_rank; in CheckInputs()
3787 if (*pack_axis < 0 || *pack_axis >= pack_output_rank) { in CheckInputs()
3798 int pack_axis, int64_t* slice_start_value, bool* found, in GetSliceAxis() argument
[all …]
|