Home
last modified time | relevance | path

Searched defs:gradients (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py73 gradients, hessians): argument
196 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
209 def schedule_add(self, partition_ids, feature_ids, gradients, hessians): argument
230 def _make_summary(self, partition_ids, feature_ids, gradients, hessians): argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc103 HloInstruction* gradients = in TEST_F() local
144 HloInstruction* gradients = in TEST_F() local
174 HloInstruction* gradients = in TEST_F() local
204 HloInstruction* gradients = in TEST_F() local
233 HloInstruction* gradients = in TEST_F() local
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dbase_split_handler.py77 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
102 def update_stats_sync(self, stamp_token, example_partition_ids, gradients, argument
Dordinal_split_handler.py223 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
400 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
607 quantile_buckets, example_partition_ids, gradients, argument
656 sparse_column_shape, quantile_buckets, example_partition_ids, gradients, argument
Dcategorical_split_handler.py93 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py619 def update_stats(self, loss, predictions_dict, gradients=None, hessians=None): argument
1179 def train(self, loss, predictions_dict, labels, gradients=None, argument
1299 gradients, argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc1151 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1170 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1190 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1211 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1234 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1260 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1287 auto gradients = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1303 auto gradients = ConstantR3FromArray3D<float>( in XLA_TEST_F() local
1319 auto gradients = in XLA_TEST_F() local
1339 auto gradients = ConstantLiteral(&builder, gradients_literal); in XLA_TEST_F() local
[all …]
/external/tensorflow/tensorflow/python/ops/
Dgradients_impl.py44 def gradients(ys, function
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc153 auto gradients = gradients_t->vec<float>(); in SerializeScalarAccumulatorToOutput() local
196 auto gradients = gradients_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput() local
231 const auto& gradients = gradients_t.vec<float>(); in AddToScalarAccumulator() local
277 const auto& gradients = gradients_t.flat_outer_dims<float>(); in AddToTensorAccumulator() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlrn_ops.cc165 xla::XlaOp gradients = xla::Add( in Compile() local
Drelu_op.cc110 auto gradients = ctx->Input("gradients"); in Compile() local
Dconv_op_helpers.cc396 const xla::Shape& filter_shape, xla::XlaOp gradients, in MakeXlaBackpropFilterConvOp()
Dpooling_ops.cc337 xla::XlaOp gradients = in Compile() local
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_loss_calculator.h112 VectorOut* loss, std::vector<MatrixOut>* gradients, in CalculateLoss()
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet_test.py151 def apply_gradients(model, optimizer, gradients): argument
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc225 const auto gradients = gradients_t->matrix<float>(); in Compute() local
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50_test.py70 def apply_gradients(model, optimizer, gradients): argument
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py542 def _aggregate_grads(gradients): argument
/external/deqp/modules/gles2/functional/
Des2fTextureSizeTests.cpp228 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local
/external/deqp/modules/gles3/functional/
Des3fTextureSizeTests.cpp229 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local
/external/deqp/modules/glshared/
DglsRandomShaderCase.cpp323 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in getTexCube() local
/external/tensorflow/tensorflow/c/eager/
Dtape.h488 gtl::FlatMap<int64, std::vector<Gradient*>> gradients; in ComputeGradient() local
/external/skqp/src/pdf/
DSkPDFGradientShader.cpp165 int gradients = 0; in gradient_function_code() local
/external/skia/src/pdf/
DSkPDFGradientShader.cpp165 int gradients = 0; in gradient_function_code() local
/external/deqp/modules/gles3/accuracy/
Des3aTextureFilteringTests.cpp398 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local

12