Searched refs:GetOrCreateMax (Results 1 – 6 of 6) sorted by relevance
56 const xla::XlaComputation& max_func = *ctx->GetOrCreateMax(type); in Compile()94 const xla::XlaComputation& max_func = *ctx->GetOrCreateMax(type); in CrossEntropyWithLogits()
68 const xla::XlaComputation* fmax = ctx->GetOrCreateMax(data_type); in Compile()
77 const xla::XlaComputation* GetOrCreateMax(const DataType type);
78 const xla::XlaComputation* XlaContext::GetOrCreateMax(const DataType type) { in GetOrCreateMax() function in tensorflow::XlaContext
248 const xla::XlaComputation* GetOrCreateMax(const DataType type);
572 const xla::XlaComputation* XlaOpKernelContext::GetOrCreateMax( in GetOrCreateMax() function in tensorflow::XlaOpKernelContext574 return xla_context()->GetOrCreateMax(type); in GetOrCreateMax()