Home
last modified time | relevance | path

Searched refs:pack_axis (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3067 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 …]