Searched refs:pack_output_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3066 PartialTensorShape pack_output_shape; 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()
3107 PartialTensorShape* pack_output_shape, int* pack_axis, in CheckInputs() argument
3119 *pack_output_shape = slice_properties[0].shape(); in CheckInputs()
3120 const int pack_input_rank = pack_output_shape->dims() - 1; in CheckInputs()
3135 const PartialTensorShape& pack_output_shape, in GetSliceAxis() argument
3191 &slice_begin_t, &slice_end_t, slice_strides_t, pack_output_shape, in GetSliceAxis()
3216 if (v != pack_output_shape.dim_size(i)) { in GetSliceAxis()
3238 (end_index == -1) ? pack_output_shape.dim_size(slice_axis) : end_value; in GetSliceAxis()
[all …]
|