Home
last modified time | relevance | path

Searched refs:CreateScalarIdentityWithZeroComputation (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.h71 XlaComputation CreateScalarIdentityWithZeroComputation(PrimitiveType type,
Darithmetic.cc91 XlaComputation CreateScalarIdentityWithZeroComputation(PrimitiveType type, in CreateScalarIdentityWithZeroComputation() function
Dmatrix.cc98 CreateScalarIdentityWithZeroComputation(shape.element_type(), builder), in GetMatrixDiagonal()
294 CreateScalarIdentityWithZeroComputation( in EinsumDiagonal()
Dslicing.cc198 CreateScalarIdentityWithZeroComputation( in TorchGather()