Searched defs:backprop (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 95 xla::XlaOp backprop = in CrossEntropyWithLogits() local
113 xla::XlaOp loss, backprop; in Compile() local
180 xla::XlaOp loss, backprop; in Compile() local
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | nn_ops.cc | 50 AbstractTensorHandle* backprop; in PYBIND11_MODULE() 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_input_ops_benchmark_test.cc | 77 Node* backprop = test::graph::Constant(graph, out_backprop_t, "backprop"); in Conv2DBackpropInput() local
|
| D | conv_grad_filter_ops_benchmark_test.cc | 78 Node* backprop = test::graph::Constant(graph, out_backprop_t, "backprop"); in Conv2DBackpropFilter() local
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | nn_ops.cc | 45 AbstractTensorHandle** backprop, in SparseSoftmaxCrossEntropyWithLogits()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | softmax_spmd_expander.cc | 468 mlir::Operation* op, const mlir::Value& loss, const mlir::Value& backprop, in MaybeRelayoutOutputs()
614 mlir::Value backprop = in ExpandOp() local
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad_test.cc | 43 AbstractTensorHandle* backprop; in SparseSoftmaxCrossEntropyWithLogitsModel() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 290 void HandleConv2DBackPropFilter(TF::Conv2DBackpropFilterOp backprop, in HandleConv2DBackPropFilter()
579 if (auto backprop = dyn_cast<TF::Conv2DBackpropFilterOp>(user)) { in HandleFirstConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.cc | 558 auto backprop = Parameter(&builder, 1, shape, "backprop"); in CreateScalarReluSensitivity() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2358 std::vector<int64_t> backprop = in TEST_F() local
2384 std::vector<int64_t> backprop = in TEST_F() local
2436 std::vector<int64_t> backprop = in TEST_F() local
2464 std::vector<int64_t> backprop = in TEST_F() local
2558 std::vector<int64_t> backprop = ImageDims( in TEST_F() local
2587 std::vector<int64_t> backprop = ImageDims( in TEST_F() local
|