Searched refs:GetAndValidateAttributes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sharding_util_ops.cc | 41 Status GetAndValidateAttributes(OpKernelConstruction* ctx, in GetAndValidateAttributes() function
133 GetAndValidateAttributes<true>(ctx, num_splits_, num_slices_, in XlaSplitNDBaseOp()
306 ctx, GetAndValidateAttributes<false>(ctx, num_concats_, num_slices_, in XlaConcatNDBaseOp()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops.cc | 106 void GetAndValidateAttributes(bool split, OpKernelConstruction* ctx, in GetAndValidateAttributes() function
412 GetAndValidateAttributes(/*split=*/true, ctx, num_splits_, num_slices_, in XlaSplitNDShared()
682 GetAndValidateAttributes(/*split=*/false, ctx, num_concats_, num_slices_, in XlaConcatNDShared()
|