Searched refs:ResolveInputDynamismIntoPred (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 86 ctx, ctx->ResolveInputDynamismIntoPred(2, &num_segments_is_dynamic)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 62 ctx->ResolveInputDynamismIntoPred(1, &leading_dim_is_dynamic)); in GetTensorListDynamicDims()
150 ctx, ctx->ResolveInputDynamismIntoPred(1, &num_element_is_dynamic)); in Compile()
220 ctx, ctx->ResolveInputDynamismIntoPred(1, &num_element_is_dynamic)); in Compile()
|
| D | categorical_op.cc | 96 ctx, ctx->ResolveInputDynamismIntoPred(1, &num_samples_is_dynamic)); in Compile()
|
| D | segment_reduction_ops.cc | 96 ctx, ctx->ResolveInputDynamismIntoPred(2, &num_segments_is_dynamic)); in Compile()
|
| D | sequence_ops.cc | 115 ctx->ResolveInputDynamismIntoPred(1, &limit_is_dynamic)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 122 Status ResolveInputDynamismIntoPred(int index, bool* out);
|
| D | xla_op_kernel.cc | 262 Status XlaOpKernelContext::ResolveInputDynamismIntoPred(int index, bool* out) { in ResolveInputDynamismIntoPred() function in tensorflow::XlaOpKernelContext
|