Searched defs:axis (Results 1 - 25 of 145) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpack_op.cc42 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
64 int axis = axis_; variable
65 if (axis < 0) axis += expanded_num_dims;
67 OP_REQUIRES(ctx, 0 <= axis && axis < expanded_num_dims,
68 errors::InvalidArgument("axis = ", axis_, " not in [",
75 child_shape.InsertDim(axis, 1);
83 ctx->SetOutput(0, ctx->builder()->ConcatInDim(reshaped_inputs, axis));
H A Dindex_ops.cc50 const int axis = dim < 0 ? dim + input_dims : dim; local
53 ctx, axis >= 0 && axis < input_dims,
56 const int64 axis_size = input_shape.dim_size(axis);
59 errors::InvalidArgument("Reduction axis ", dim, " is empty in shape ",
71 index_type, axis, &output));
75 index_type, axis, &output));
H A Done_hot_op.cc29 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
44 errors::InvalidArgument("Expected axis to be -1 or between [0, ",
56 const int axis = (axis_ == -1) ? indices_dims : axis_; variable
67 ctx, XlaHelpers::OneHot(ctx->builder(), depth, axis, input_type(0),
H A Dunpack_op.cc42 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
49 int axis = axis_; variable
50 if (axis < 0) axis += input_shape.dims();
52 OP_REQUIRES(ctx, 0 <= axis && axis < input_shape.dims(),
53 errors::InvalidArgument("axis = ", axis_, " not in [",
58 ctx, input_shape.dims() > 0 && input_shape.dim_size(axis) == num,
59 errors::InvalidArgument("Input shape axis ", axis, " mus
[all...]
H A Dscan_ops.cc55 errors::InvalidArgument("ScanOp: axis must be a scalar, not ",
58 int64 axis; variable
59 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &axis));
60 if (axis < 0) {
61 axis += input_shape.dims();
64 ctx, FastBoundsCheck(axis, input_shape.dims()),
65 errors::InvalidArgument("ScanOp: Expected scan axis in the range [",
67 "), but got ", axis));
81 window_dims[axis] = input_shape.dim_size(axis);
[all...]
H A Dconcat_op.cc64 int32 axis = concat_dim < 0 ? concat_dim + input_dims : concat_dim; variable
66 (0 <= axis && axis < input_dims) ||
95 output_concat_dim += in_shape.dims() > 0 ? in_shape.dim_size(axis) : 1;
98 VLOG(1) << "Concat dim " << concat_dim << " equivalent to " << axis;
99 ctx->SetOutput(0, ctx->builder()->ConcatInDim(input_data, axis));
113 // is the last input instead of the first and renamed to 'axis'.
123 .CompileTimeConstInput("axis"),
172 int32 axis = cdim < 0 ? cdim + dims : cdim; variable
173 OP_REQUIRES(ctx, FastBoundsCheck(axis, dim
[all...]
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
H A Dreduce_slice_ops.cc31 // "axis" must be a scala
41 // if "indices" is a vector of 0 elements, then the axis dimension of
57 int64 axis = _axis->scalar<int64>()(); local
58 TF_RETURN_IF_ERROR(c->ReplaceDim(handle, axis, dim_axis, &handle));
69 .Input("axis: int64")
114 .Input("axis: int64")
171 .Input("axis: int64")
228 .Input("axis: int64")
/external/tensorflow/tensorflow/core/kernels/
H A Dunique_op.cc47 int64 axis = 0; variable
53 // In case of UniqueV2, the axis is a 1D vector. The purpose is
54 // to allow specifying either "no axis" or "axis". The `[]` means
55 // "no axis", while `[x]` means `axis = x`.
58 errors::InvalidArgument("axis expects a 1D vector."));
62 "axis does not support input tensors larger than 1 elements"));
71 "axis tensor should be int32 or int64, but got ",
74 axis
[all...]
H A Dreduction_ops_common.cc61 Status SimplifyHelper(const Tensor& data, const Tensor& axis, argument
63 auto axis_vec = axis.flat<Tperm>();
64 for (int64 i = 0; i < axis.NumElements(); ++i) {
77 Status ReductionHelper::Simplify(const Tensor& data, const Tensor& axis, argument
79 // bitmap[i] indicates whether to reduce data along i-th axis.
81 if (axis.dtype() == DT_INT32) {
82 TF_RETURN_IF_ERROR(SimplifyHelper<int32>(data, axis, bitmap));
84 TF_RETURN_IF_ERROR(SimplifyHelper<int64>(data, axis, bitmap));
100 // Depending on bitmap[i] and bitmap[i-1], we can collapse axis of
/external/apache-xml/src/main/java/org/apache/xpath/patterns/
H A DContextMatchStepPattern.java41 public ContextMatchStepPattern(int axis, int paxis) argument
43 super(DTMFilter.SHOW_ALL, axis, paxis);
99 int axis = m_axis;
101 boolean needToTraverseAttrs = WalkerFactory.isDownwardAxisOfMany(axis);
105 if((Axis.PRECEDING == axis) && iterRootIsAttr)
107 axis = Axis.PRECEDINGANDANCESTOR;
110 traverser = dtm.getAxisTraverser(axis);
H A DFunctionPattern.java46 public FunctionPattern(Expression expr, int axis, int predaxis) argument
49 super(0, null, null, axis, predaxis);
/external/droiddriver/src/io/appium/droiddriver/scroll/
H A DStepBasedScroller.java44 private final Axis axis; field in class:StepBasedScroller
54 * @param axis the axis this scroller can scroll
61 public StepBasedScroller(int maxScrolls, long perScrollTimeoutMillis, Axis axis, argument
65 this.axis = axis;
72 * perScrollTimeoutMillis, vertical axis, not startFromBegining.
133 PhysicalDirection backwardDirection = converter.toPhysicalDirection(axis, BACKWARD);
/external/eigen/demos/opengl/
H A Dtrackball.cpp24 Vector3f axis = mLastPoint3D.cross(newPoint3D).normalized(); local
30 mpCamera->rotateAroundTarget(Quaternionf(AngleAxisf(angle, axis)));
32 mpCamera->localRotate(Quaternionf(AngleAxisf(-angle, axis)));
/external/guava/guava/src/com/google/common/collect/
H A DCartesianList.java69 private int getAxisIndexForProductIndex(int index, int axis) { argument
70 return (index / axesSizeProduct[axis + 1]) % axes.get(axis).size();
84 public E get(int axis) {
85 checkElementIndex(axis, size());
86 int axisIndex = getAxisIndexForProductIndex(index, axis);
87 return axes.get(axis).get(axisIndex);
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dsqueeze_test.cc30 std::initializer_list<int> axis) {
35 CreateSqueezeOptions(builder_, builder_.CreateVector<int>(axis))
29 BaseSqueezeOpModel(const TensorData& input, const TensorData& output, std::initializer_list<int> axis) argument
H A Dconcatenation_test.cc29 // TODO(ahentz): Also test different activation types, axis, input
31 BaseConcatenationOpModel(const TensorData& input_template, int axis, argument
42 CreateConcatenationOptions(builder_, axis, ActivationFunctionType_NONE)
74 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/1,
82 ConcatenationOpModel m0({TensorType_FLOAT32, {1}}, /*axis=*/0,
90 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 3}}, /*axis=*/0,
102 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 3}}, /*axis=*/0,
110 ConcatenationOpModel m0_negative({TensorType_FLOAT32, {2, 3}}, /*axis=*/-2,
118 ConcatenationOpModel m1({TensorType_FLOAT32, {2, 3}}, /*axis=*/1,
126 ConcatenationOpModel m1_negative({TensorType_FLOAT32, {2, 3}}, /*axis
[all...]
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dconvert_expanddims_to_reshape.cc51 // Yield until input axis array shape has been resolved.
56 // Yield until the input axis array is constant
59 int axis = axis_array.GetBuffer<ArrayDataType::kInt32>().data[0]; local
61 if (axis < 0) {
62 axis = reshape_dims.size();
64 reshape_dims.insert(reshape_dims.begin() + axis, 1);
66 // The input tensor has shape, and the axis input is constant. We can now
85 // Delete axis array if unused
H A Dresolve_tensorflow_concat.cc38 // of inputs: in Concat,the axis is the first input, while in
51 // If the axis array hasn't been resolved to a constant yet,
55 AddMessageF("Waiting for the axis of %s to be resolved to a constant",
63 const int axis = axis_data[0]; local
67 concatenation_op->axis = axis;
76 // Remove the axis array if it is not used by anything else.
/external/apache-xml/src/main/java/org/apache/xpath/axes/
H A DOneStepIteratorForward.java40 /** The traversal axis from where the nodes will be filtered. */
65 * @param axis One of the org.apache.xml.dtm.Axis integers.
69 public OneStepIteratorForward(int axis) argument
73 m_axis = axis;
147 * Returns the axis being iterated, if it is known.
149 * @return Axis.CHILD, etc., or -1 if the axis is not known or is of multiple
H A DReverseAxesWalker.java40 ReverseAxesWalker(LocPathIterator locPathIterator, int axis) argument
42 super(locPathIterator, axis);
245 /** The DTM inner traversal class, that corresponds to the super axis. */
/external/freetype/include/freetype/
H A Dftmm.h44 /* setting design axis coordinates. */
60 /* A structure to model a given axis in design space for Multiple */
67 /* name :: The axis's name. */
69 /* minimum :: The axis's minimum design coordinate. */
71 /* maximum :: The axis's maximum design coordinate. */
102 /* axis :: A table of axis descriptors. */
108 FT_MM_Axis axis[T1_MAX_MM_AXIS]; member in struct:FT_Multi_Master_
119 /* A structure to model a given axis in design space for Multiple */
123 /* name :: The axis'
242 FT_Var_Axis* axis; member in struct:FT_MM_Var_
[all...]
/external/skia/include/core/
H A DSkFontArguments.h18 SkFourByteTag axis; member in struct:SkFontArguments::VariationPosition::Coordinate
52 * Any axis not specified will use the default value.
53 * Any specified axis not actually present in the font will be ignored.
/external/skia/src/core/
H A DSkFontDescriptor.cpp59 SkAutoSTMalloc<4, SkFixed> axis; local
75 axis.reset(axisCount);
77 axis[i] = read_uint(stream);
94 SkMemoryStream::Make(std::move(data)), index, axis, axisCount);
/external/skqp/include/core/
H A DSkFontArguments.h18 SkFourByteTag axis; member in struct:SkFontArguments::VariationPosition::Coordinate
52 * Any axis not specified will use the default value.
53 * Any specified axis not actually present in the font will be ignored.
/external/skqp/src/core/
H A DSkFontDescriptor.cpp59 SkAutoSTMalloc<4, SkFixed> axis; local
75 axis.reset(axisCount);
77 axis[i] = read_uint(stream);
94 SkMemoryStream::Make(std::move(data)), index, axis, axisCount);

Completed in 2136 milliseconds

123456