Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc42 if (!ctx->ValidateInputsAreSameShape(this)) return; in Compute()
157 if (!ctx->ValidateInputsAreSameShape(this)) return; in Compute()
Dcwise_op_select.cc113 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.h826 bool ValidateInputsAreSameShape(OpKernel* op);
Dop_kernel.cc1016 bool OpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { in ValidateInputsAreSameShape() function in tensorflow::OpKernelContext
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h112 bool ValidateInputsAreSameShape(OpKernel* op) TF_MUST_USE_RESULT;
Dxla_op_kernel.cc36 bool XlaOpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { in ValidateInputsAreSameShape() function in tensorflow::XlaOpKernelContext
37 return context_->ValidateInputsAreSameShape(op); in ValidateInputsAreSameShape()