Home
last modified time | relevance | path

Searched refs:AnalyzeInputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlinalg_ops_common.cc94 AnalyzeInputs(context, &inputs, &input_matrix_shapes, &batch_shape); in Compute()
117 void LinearAlgebraOp<InputScalar, OutputScalar>::AnalyzeInputs( in AnalyzeInputs() function in tensorflow::LinearAlgebraOp
Dlinalg_ops_common.h179 void AnalyzeInputs(OpKernelContext* context, TensorInputs* inputs,
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc446 Status AnalyzeInputs(ScopedAllocatorOptimizer* sa_opti, in AnalyzeInputs() function in tensorflow::grappler::UnaryElementwiseRewriter
847 TF_RETURN_IF_ERROR(AnalyzeInputs( in Rewrite()