Searched refs:GetInputsAndOutputShape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sharding_util_ops.cc | 319 TF_RETURN_IF_ERROR(GetInputsAndOutputShape(ctx, input_handles, input_shapes, in CompileInternal()
397 Status GetInputsAndOutputShape(XlaOpKernelContext* ctx, in GetInputsAndOutputShape() function in tensorflow::__anone83cea6a0111::XlaConcatNDBaseOp
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops.cc | 687 Status GetInputsAndOutputShape(OpKernelContext* ctx, OpInputList& inputs, in GetInputsAndOutputShape() function in tensorflow::__anonda0888800111::XlaConcatNDShared
857 this->GetInputsAndOutputShape(ctx, inputs, output_shape)); in Compute()
887 this->GetInputsAndOutputShape(ctx, inputs, output_shape)); in Compute()
|