Home
last modified time | relevance | path

Searched defs:backprop (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dsoftmax_op.cc95 xla::XlaOp backprop = in CrossEntropyWithLogits() local
113 xla::XlaOp loss, backprop; in Compile() local
180 xla::XlaOp loss, backprop; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drelu_op_gpu.cu.cc46 Eigen::half* __restrict__ backprop, in ReluGradHalfKernel()
99 const Eigen::half* __restrict__ feature, Eigen::half* __restrict__ backprop, in ReluGradHalfKernelVector()
H A Dconv_grad_filter_ops_benchmark_test.cc78 Node* backprop = test::graph::Constant(graph, out_backprop_t, "backprop"); in Conv2DBackpropFilter() local
H A Dconv_grad_input_ops_benchmark_test.cc77 Node* backprop = test::graph::Constant(graph, out_backprop_t, "backprop"); in Conv2DBackpropInput() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsoftmax_spmd_expander.cc468 mlir::Operation* op, const mlir::Value& loss, const mlir::Value& backprop, in MaybeRelayoutOutputs()
614 mlir::Value backprop = in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dnn_ops.cc50 AbstractTensorHandle* backprop; in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_space_to_depth_pass.cc290 void HandleConv2DBackPropFilter(TF::Conv2DBackpropFilterOp backprop, in HandleConv2DBackPropFilter()
579 if (auto backprop = dyn_cast<TF::Conv2DBackpropFilterOp>(user)) { in HandleFirstConvolution() local
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dnn_ops.cc45 AbstractTensorHandle** backprop, in SparseSoftmaxCrossEntropyWithLogits()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Dnn_grad_test.cc43 AbstractTensorHandle* backprop; in SparseSoftmaxCrossEntropyWithLogitsModel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc2358 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dclient_library_test_base.cc558 auto backprop = Parameter(&builder, 1, shape, "backprop"); in CreateScalarReluSensitivity() local