Searched defs:backprop (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 166 xla::XlaOp backprop = in CrossEntropyWithLogits() local
184 xla::XlaOp loss, backprop; in Compile() local
251 xla::XlaOp loss, backprop; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_gpu.cu.cc | 46 Eigen::half* __restrict__ backprop, in ReluGradHalfKernel()
99 const Eigen::half* __restrict__ feature, Eigen::half* __restrict__ backprop, in ReluGradHalfKernelVector()
|
| D | conv_grad_filter_ops_benchmark_test.cc | 78 Node* backprop = test::graph::Constant(graph, out_backprop_t, "backprop"); in Conv2DBackpropFilter() local
|
| D | conv_grad_input_ops_benchmark_test.cc | 77 Node* backprop = test::graph::Constant(graph, out_backprop_t, "backprop"); in Conv2DBackpropInput() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 335 void HandleConv2DBackPropFilter(TF::Conv2DBackpropFilterOp backprop, in HandleConv2DBackPropFilter()
624 if (auto backprop = dyn_cast<TF::Conv2DBackpropFilterOp>(user)) { in HandleFirstConvolution() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1630 std::vector<int64> backprop = in TEST_F() local
1656 std::vector<int64> backprop = in TEST_F() local
1704 std::vector<int64> backprop = in TEST_F() local
1730 std::vector<int64> backprop = in TEST_F() local
1809 std::vector<int64> backprop = ImageDims( in TEST_F() local
1836 std::vector<int64> backprop = ImageDims( in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.cc | 556 auto backprop = Parameter(&builder, 1, shape, "backprop"); in CreateScalarReluSensitivity() local
|