| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A 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()
|
| H A D | pooling_ops.cc | 102 auto status = ctx->ConstantInputAsIntVector(1, &ksize); in GetKernelSize()
126 auto status = ctx->ConstantInputAsIntVector(2, &stride); in GetStride()
344 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
351 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
568 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
575 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
|
| H A 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()
|
| H A D | bcast_ops.cc | 49 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector( in Compile()
105 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector( in Compile()
|
| H A D | transpose_op.cc | 57 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("perm", &perm)); in Compile()
164 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(0, &perm)); in InvertPermutation()
|
| H A D | concat_op.cc | 159 ctx->ConstantInputAsIntVector( in Compile()
180 ctx, ctx->ConstantInputAsIntVector( in Compile()
|
| H A D | slice_op.cc | 63 ctx->ConstantInputAsIntVector(1, &begin).ok(); in Compile()
65 ctx->ConstantInputAsIntVector(2, &size).ok(); in Compile()
|
| H A D | tensor_array_ops.cc | 317 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); in Compile()
388 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); in Compile()
491 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &lengths)); in Compile()
|
| H A D | empty_op.cc | 50 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("shape", &shape)); in Compile()
|
| H A D | fill_op.cc | 50 ctx->ConstantInputAsIntVector( in Compile()
|
| H A D | gather_scatter_ops.cc | 40 ctx->ConstantInputAsIntVector("slice_sizes", &slice_sizes)); in Compile()
|
| H A D | tile_ops.cc | 68 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector( in Compile()
|
| H A D | reverse_op.cc | 93 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &axes)); in Compile()
|
| H A D | xla_broadcast_helper_op.cc | 46 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("broadcast_dims", in Compile()
|
| H A D | tensor_list_ops.cc | 57 TF_RETURN_IF_ERROR(ctx->ConstantInputAsIntVector(0, &dynamic_sizes)); in GetTensorListDynamicDims()
548 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &lengths)); in Compile()
|
| H A D | fft_ops.cc | 56 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &fft_length)); in Compile()
|
| H A D | reshape_op.cc | 50 ctx->ConstantInputAsIntVector( in Compile()
|
| H A D | split_op.cc | 148 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &split_sizes)); in Compile()
|
| H A D | batchtospace_op.cc | 152 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| H A D | spacetobatch_op.cc | 168 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| H A D | unique_op.cc | 268 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &axises)); in Compile()
|
| H A D | matrix_diag_ops.cc | 71 context->ConstantInputAsIntVector("k", &diag_index)); in ProcessDiagIndex()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_kernel.h | 171 Status ConstantInputAsIntVector(
174 Status ConstantInputAsIntVector(
|
| H A D | xla_op_kernel.cc | 376 Status XlaOpKernelContext::ConstantInputAsIntVector( in ConstantInputAsIntVector() function in tensorflow::XlaOpKernelContext
383 Status XlaOpKernelContext::ConstantInputAsIntVector( in ConstantInputAsIntVector() function in tensorflow::XlaOpKernelContext
387 return ConstantInputAsIntVector(index, out, mode); in ConstantInputAsIntVector()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | image_resize_ops.cc | 41 auto status = ctx->ConstantInputAsIntVector(1, &out_size); in GetOutputShape()
|