| /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_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 | 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 | fill_op.cc | 48 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("dims", &dims)); in Compile()
53 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("dims", &dynamic_dims)); in Compile()
|
| D | tile_ops.cc | 66 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("multiples", &multiples)); in Compile()
85 ctx, ctx->ConstantInputAsIntVector("multiples", &dynamic_multiples)); in Compile()
|
| D | slice_op.cc | 61 ctx->ConstantInputAsIntVector(1, &begin).ok(); in Compile()
62 const bool size_is_constant = ctx->ConstantInputAsIntVector(2, &size).ok(); in Compile()
106 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &dynamic_size)); 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 | 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 | 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 | transpose_op.cc | 57 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("perm", &perm)); in Compile()
163 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(0, &perm)); in InvertPermutation()
|
| D | pooling_ops.cc | 94 auto status = ctx->ConstantInputAsIntVector(1, &ksize); in GetKernelSize()
118 auto status = ctx->ConstantInputAsIntVector(2, &stride); in GetStride()
296 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
303 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
516 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
523 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); 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 | empty_op.cc | 50 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("shape", &shape)); in Compile()
|
| D | gather_scatter_ops.cc | 40 ctx->ConstantInputAsIntVector("slice_sizes", &slice_sizes)); 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 | dynamic_slice_ops.cc | 110 ctx, ctx->ConstantInputAsIntVector("size_indices", &size_indices)); 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 | reshape_op.cc | 48 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &shape_input)); in Compile()
|
| D | fft_ops.cc | 55 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &fft_length)); in Compile()
|
| D | batchtospace_op.cc | 152 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| D | split_op.cc | 147 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &split_sizes)); in Compile()
|
| D | spacetobatch_op.cc | 154 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 139 Status ConstantInputAsIntVector(int index, std::vector<int64>* out);
140 Status ConstantInputAsIntVector(absl::string_view name,
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 41 auto status = ctx->ConstantInputAsIntVector(1, &out_size); in GetOutputShape()
|