Searched refs:axis_arg (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scan_ops.cc | 52 const Tidx axis_arg = in Compute() local
54 const Tidx axis = (axis_arg < 0) ? input.dims() + axis_arg : axis_arg; in Compute()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops_test.py | 694 axis_arg = fn2(axis) if axis is not None else None
696 np_array_ops.amax(arr_arg, axis=axis_arg, *args, **kwargs),
699 np_array_ops.amin(arr_arg, axis=axis_arg, *args, **kwargs),
723 axis_arg = fn2(axis) if axis is not None else None
725 np_array_ops.mean(arr_arg, axis=axis_arg, *args, **kwargs),
782 axis_arg = fn2(axis) if axis is not None else None
784 math_fun(arg1, axis=axis_arg, **kwargs),
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 3674 int axis_arg = -1; in PartialConcatConstFolding() local
3679 axis_arg = 0; in PartialConcatConstFolding()
3682 axis_arg = num_non_control_inputs - 1; in PartialConcatConstFolding()
3739 added_node->add_input(node->input(axis_arg)); in PartialConcatConstFolding()
3741 node_map_->AddOutput(NodeName(node->input(axis_arg)), added_node->name()); in PartialConcatConstFolding()
|