Searched refs:GetOrCreateMax (Results 1 – 6 of 6) sorted by relevance
88 const xla::XlaComputation* GetOrCreateMax(const DataType type);
85 const xla::XlaComputation* XlaContext::GetOrCreateMax(const DataType type) { in GetOrCreateMax() function in tensorflow::XlaContext
326 const xla::XlaComputation* GetOrCreateMax(const DataType type);
775 const xla::XlaComputation* XlaOpKernelContext::GetOrCreateMax( in GetOrCreateMax() function in tensorflow::XlaOpKernelContext777 return xla_context()->GetOrCreateMax(type); in GetOrCreateMax()
48 const xla::XlaComputation& max_func = *ctx->GetOrCreateMax(type); in CrossEntropyWithLogits()
73 const xla::XlaComputation* fmax = ctx->GetOrCreateMax(data_type); in Compile()