Searched refs:ResolveInputDynamismIntoPred (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sequence_ops.cc | 120 ctx->ResolveInputDynamismIntoPred(0, &start_is_dynamic)); in Compile()
123 ctx->ResolveInputDynamismIntoPred(1, &limit_is_dynamic)); in Compile()
|
| H A D | tensor_list_ops.cc | 63 ctx->ResolveInputDynamismIntoPred(1, &leading_dim_is_dynamic)); in GetTensorListDynamicDims()
154 ctx, ctx->ResolveInputDynamismIntoPred(1, &num_element_is_dynamic)); in Compile()
226 ctx, ctx->ResolveInputDynamismIntoPred(1, &num_element_is_dynamic)); in Compile()
|
| H A D | topk_op.cc | 66 context->ResolveInputDynamismIntoPred(1, &k_is_dynamic)); in Compile()
|
| H A D | categorical_op.cc | 71 ctx, ctx->ResolveInputDynamismIntoPred(1, &num_samples_is_dynamic)); in Compile()
|
| H A D | segment_reduction_ops.cc | 98 ctx, ctx->ResolveInputDynamismIntoPred(2, &num_segments_is_dynamic)); in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_kernel.h | 131 Status ResolveInputDynamismIntoPred(int index, bool* out);
134 Status ResolveInputDynamismIntoPred(absl::string_view name, bool* out);
|
| H A D | xla_op_kernel.cc | 270 Status XlaOpKernelContext::ResolveInputDynamismIntoPred(int index, bool* out) { in ResolveInputDynamismIntoPred() function in tensorflow::XlaOpKernelContext
306 Status XlaOpKernelContext::ResolveInputDynamismIntoPred(absl::string_view name, in ResolveInputDynamismIntoPred() function in tensorflow::XlaOpKernelContext
309 return ResolveInputDynamismIntoPred(index, out); in ResolveInputDynamismIntoPred()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | segment_reduction_ops.cc | 88 ctx, ctx->ResolveInputDynamismIntoPred(2, &num_segments_is_dynamic)); in Compile()
|