Searched defs:backprop (Results 1 – 4 of 4) sorted by relevance
137 xla::XlaOp backprop = in CrossEntropyWithLogits() local165 xla::XlaOp loss, backprop; in Compile() local232 xla::XlaOp loss, backprop; in Compile() local
39 Eigen::half* backprop, int32 count) { in ReluGradHalfKernel()
1629 std::vector<int64> backprop = in TEST_F() local1655 std::vector<int64> backprop = in TEST_F() local1703 std::vector<int64> backprop = in TEST_F() local1729 std::vector<int64> backprop = in TEST_F() local1808 std::vector<int64> backprop = ImageDims( in TEST_F() local1835 std::vector<int64> backprop = ImageDims( in TEST_F() local
556 auto backprop = Parameter(&builder, 1, shape, "backprop"); in CreateScalarReluSensitivity() local