Searched refs:ConstantInputAsShape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 49 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape( in Compile()
86 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
126 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape( in Compile()
158 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
217 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | stateless_random_ops.cc | 154 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
199 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
250 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
291 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
342 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
386 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | stateless_random_ops_v2.cc | 218 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape( in Compile()
252 Status status = ctx->ConstantInputAsShape(0, &static_shape); in Compile()
288 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
352 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
401 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape( in Compile()
436 Status status = ctx->ConstantInputAsShape(0, &static_shape); in Compile()
472 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &shape)); in Compile()
|
| D | conv_ops.cc | 108 ctx, ctx->ConstantInputAsShape(0, &input_tensor_shape, in Compile()
176 ctx, ctx->ConstantInputAsShape(1, &filter_tensor_shape, in Compile()
|
| D | broadcast_to_op.cc | 34 context->ConstantInputAsShape( in Compile()
|
| D | sparse_to_dense_op.cc | 45 context->ConstantInputAsShape( in Compile()
|
| D | scatter_nd_op.cc | 96 OP_REQUIRES_OK(context, context->ConstantInputAsShape(2, &buffer_shape)); in Compile()
|
| D | stateful_random_ops.cc | 196 TF_RETURN_IF_ERROR(ctx->ConstantInputAsShape(shape_input_idx, &shape)); in CompileImpl()
|
| D | pooling_ops.cc | 473 ctx, ctx->ConstantInputAsShape(0, &gradients_shape, in Compile()
|
| D | strided_slice_op.cc | 523 ctx, ctx->ConstantInputAsShape(0, &input_shape, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 196 Status ConstantInputAsShape(
|
| D | xla_op_kernel.cc | 441 Status XlaOpKernelContext::ConstantInputAsShape(int index, TensorShape* shape, in ConstantInputAsShape() function in tensorflow::XlaOpKernelContext
|