Searched refs:pack_axis (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | arithmetic_optimizer.cc | 3067 int pack_axis; in TrySimplify() local 3069 CheckInputs(node, pack, &pack_output_shape, &pack_axis, &return_early)); in TrySimplify() 3074 TF_RETURN_IF_ERROR(GetSliceAxis(node, pack, pack_output_shape, pack_axis, in TrySimplify() 3078 return RewriteGraph(node, pack, slice_start_value, pack_axis, in TrySimplify() 3107 PartialTensorShape* pack_output_shape, int* pack_axis, in CheckInputs() argument 3112 *pack_axis = pack->attr().at("axis").i(); in CheckInputs() 3121 if (*pack_axis < 0) { in CheckInputs() 3123 *pack_axis += pack_input_rank; in CheckInputs() 3125 if (*pack_axis < 0 || *pack_axis >= pack_input_rank) { in CheckInputs() 3136 int pack_axis, int* slice_start_value, bool* found) { in GetSliceAxis() argument [all …]
|