Home
last modified time | relevance | path

Searched refs:ValidateInputsAreSameShape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc47 if (!ctx->ValidateInputsAreSameShape(this)) return; in Compute()
162 if (!ctx->ValidateInputsAreSameShape(this)) return; in Compute()
Dcwise_op_select.cc119 if (!ctx->ValidateInputsAreSameShape(this)) return; in ComputeElementwise()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Daggregate_ops.cc30 if (!ctx->ValidateInputsAreSameShape(this)) return; in Compile()
/external/tensorflow/tensorflow/core/framework/
Dnumeric_op.h76 if (!context->ValidateInputsAreSameShape(this)) { in Compute()
Dop_kernel.h795 bool ValidateInputsAreSameShape(OpKernel* op);
Dop_kernel.cc857 bool OpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { in ValidateInputsAreSameShape() function in tensorflow::OpKernelContext
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h100 bool ValidateInputsAreSameShape(OpKernel* op) TF_MUST_USE_RESULT;
Dxla_op_kernel.cc35 bool XlaOpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { in ValidateInputsAreSameShape() function in tensorflow::XlaOpKernelContext
36 return context_->ValidateInputsAreSameShape(op); in ValidateInputsAreSameShape()