| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pack_op.cc | 42 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
64 int axis = axis_;
68 errors::InvalidArgument("axis = ", axis_, " not in [",
87 int axis_; member in class:tensorflow::__anon38759::PackOp
|
| unpack_op.cc | 42 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
49 int axis = axis_;
53 errors::InvalidArgument("axis = ", axis_, " not in [",
86 int axis_; member in class:tensorflow::__anon38791::UnpackOp
|
| one_hot_op.cc | 29 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
43 ctx, axis_ == -1 || (axis_ >= 0 && axis_ < output_dims),
45 output_dims, "). But received: ", axis_));
56 const int axis = (axis_ == -1) ? indices_dims : axis_;
74 int32 axis_; member in class:tensorflow::__anon38758::OneHotOp
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| mean_test.cc | 28 void SetAxis(std::initializer_list<int> data) { PopulateTensor(axis_, data); }
44 int axis_; member in class:tflite::__anon39265::BaseMeanOpModel
55 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape);
69 axis_ = AddInput(axis);
|
| split_test.cc | 33 axis_ = AddInput({TensorType_INT32, {1}});
35 axis_ = AddConstInput(TensorType_INT32, {axis}, {1});
44 BuildInterpreter({GetShape(axis_), GetShape(input_)});
53 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); }
62 int axis_; member in class:tflite::__anon39277::SplitOpModel
|
| /external/tensorflow/tensorflow/core/kernels/ |
| pack_op.cc | 49 OP_REQUIRES_OK(context, context->GetAttr("axis", &axis_));
67 int axis = axis_;
71 errors::InvalidArgument("axis = ", axis_, " not in [",
132 int axis_; member in class:tensorflow::PackOp
|
| unpack_op.cc | 43 OP_REQUIRES_OK(context, context->GetAttr("axis", &axis_));
51 int axis = axis_;
55 errors::InvalidArgument("axis = ", axis_, " not in [",
126 int axis_; member in class:tensorflow::UnpackOp
|
| one_hot_op.cc | 47 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_));
62 ctx, axis_ == -1 || (axis_ >= 0 && axis_ < output_dims),
64 output_dims, "). But received: ", axis_));
75 const int axis = (axis_ == -1) ? indices_dims : axis_;
123 int32 axis_; member in class:tensorflow::OneHotOp
|
| /external/annotation-tools/asmx/test/lib/ |
| jd.xslt-1.5.5.jar | |
