Searched refs: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
165 pack_axis.push_back(imaged_original_shape_current_axis + 1); in Run()
178 if (pack_axis.empty()) { in Run()
228 for (int i = pack_axis.size() - 1; i >= 0; --i) { in Run()
235 int axis = pack_axis.at(i); in Run()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3167 int pack_axis; in TrySimplify() local
3169 CheckInputs(node, pack, &pack_output_shape, &pack_axis, &return_early)); in TrySimplify()
3175 TF_RETURN_IF_ERROR(GetSliceAxis(node, pack, pack_output_shape, pack_axis, in TrySimplify()
3180 return RewriteGraph(node, pack, slice_start_value, pack_axis, in TrySimplify()
3186 PartialTensorShape* pack_output_shape, int* pack_axis, in CheckInputs() argument
3191 *pack_axis = pack->attr().at("axis").i(); in CheckInputs()
3200 if (*pack_axis < 0) { in CheckInputs()
3201 *pack_axis += pack_output_rank; in CheckInputs()
3203 if (*pack_axis < 0 || *pack_axis >= pack_output_rank) { in CheckInputs()
3214 int pack_axis, int64* slice_start_value, bool* found, in GetSliceAxis() argument
[all …]
|