| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduce_window_op.cc | 45 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
47 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("window_strides", in Compile()
49 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("base_dilations", in Compile()
51 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
|
| D | xla_conv_op.cc | 51 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("window_strides", in Compile()
53 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("lhs_dilation", in Compile()
55 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("rhs_dilation", in Compile()
|
| D | xla_pad_op.cc | 40 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("padding_low", in Compile()
42 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("padding_high", in Compile()
44 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
|
| D | listdiff_op.cc | 74 TF_RETURN_IF_ERROR(context->ConstantInputAsIntVector(0, &x_input)); in ListDiff()
75 TF_RETURN_IF_ERROR(context->ConstantInputAsIntVector(1, &y_input)); in ListDiff()
|
| D | slice_op.cc | 60 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &size)); in Compile()
61 if (ctx->ConstantInputAsIntVector(1, &begin).ok()) { in Compile()
|
| D | pooling_ops.cc | 92 auto status = ctx->ConstantInputAsIntVector(1, &ksize); in GetKernelSize()
116 auto status = ctx->ConstantInputAsIntVector(2, &stride); in GetStride()
289 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
296 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
496 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
503 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
|
| D | bcast_ops.cc | 49 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(i, &shape)); in Compile()
98 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(i, &vec)); in Compile()
|
| D | transpose_op.cc | 56 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("perm", &perm)); in Compile()
141 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(0, &perm)); in Compile()
|
| D | xla_select_and_scatter_op.cc | 44 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
46 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("window_strides", in Compile()
|
| D | concat_op.cc | 158 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &inp0_dims)); in Compile()
177 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1 + i, &inp_dims)); in Compile()
|
| D | fill_op.cc | 48 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("dims", &dims)); in Compile()
|
| D | empty_op.cc | 50 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("shape", &shape)); in Compile()
|
| D | reshape_op.cc | 46 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &shape_input)); in Compile()
|
| D | dynamic_slice_ops.cc | 98 ctx, ctx->ConstantInputAsIntVector("size_indices", &size_indices)); in Compile()
|
| D | tile_ops.cc | 66 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("multiples", &multiples)); in Compile()
|
| D | reverse_op.cc | 93 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &axes)); in Compile()
|
| D | xla_broadcast_helper_op.cc | 46 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("broadcast_dims", in Compile()
|
| D | tensor_array_ops.cc | 316 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); in Compile()
387 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); in Compile()
490 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &lengths)); in Compile()
|
| D | fft_ops.cc | 55 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &fft_length)); in Compile()
|
| D | split_op.cc | 143 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &split_sizes)); in Compile()
|
| D | batchtospace_op.cc | 152 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| D | spacetobatch_op.cc | 154 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| D | image_resize_ops.cc | 467 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &out_size)); in GeneralCompile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 124 Status ConstantInputAsIntVector(int index, std::vector<int64>* out);
125 Status ConstantInputAsIntVector(absl::string_view name,
|
| D | xla_op_kernel.cc | 250 Status XlaOpKernelContext::ConstantInputAsIntVector(int index, in ConstantInputAsIntVector() function in tensorflow::XlaOpKernelContext
257 Status XlaOpKernelContext::ConstantInputAsIntVector(absl::string_view name, in ConstantInputAsIntVector() function in tensorflow::XlaOpKernelContext
260 return ConstantInputAsIntVector(index, out); in ConstantInputAsIntVector()
|