Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py114 def add(self, stamp_token, partition_ids, feature_ids, gradients, hessians): argument
127 def schedule_add(self, partition_ids, feature_ids, gradients, hessians): argument
148 def _make_summary(self, partition_ids, feature_ids, gradients, hessians): argument
157 gradients, hessians): argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_convolution_rewriter_test.cc100 HloInstruction* gradients = in TEST_F() local
129 HloInstruction* gradients = in TEST_F() local
158 HloInstruction* gradients = in TEST_F() local
188 HloInstruction* gradients = in TEST_F() local
216 HloInstruction* gradients = in TEST_F() local
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dordinal_split_handler.py198 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
338 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
424 quantile_buckets, example_partition_ids, gradients, argument
460 sparse_column_shape, quantile_buckets, example_partition_ids, gradients, argument
Dbase_split_handler.py71 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
96 def update_stats_sync(self, stamp_token, example_partition_ids, gradients, argument
Dcategorical_split_handler.py84 def update_stats(self, stamp_token, example_partition_ids, gradients, argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc1152 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1171 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1192 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1213 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1236 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1262 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1289 auto gradients = builder.ConstantR4FromArray4D<float>( in XLA_TEST_F() local
1305 auto gradients = builder.ConstantR3FromArray3D<float>( in XLA_TEST_F() local
1321 auto gradients = in XLA_TEST_F() local
1341 auto gradients = builder.ConstantLiteral(*gradients_literal); in XLA_TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlrn_ops.cc155 xla::ComputationDataHandle gradients = builder->Add( in Compile() local
Dpooling_ops.cc347 xla::ComputationDataHandle gradients = ctx->builder()->SelectAndScatter( in Compile() local
Dconv_ops.cc547 xla::ComputationDataHandle gradients = ctx->Input(2); in Compile() local
/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/core/util/ctc/
Dctc_loss_calculator.h112 VectorOut* loss, std::vector<MatrixOut>* gradients, in CalculateLoss()
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py879 def _make_update_bias_stats_fn(self, ensemble_stamp, predictions, gradients, argument
/external/tensorflow/tensorflow/python/ops/
Dgradients_impl.py400 def gradients(ys, function
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py711 def _aggregate_grads(gradients): argument
/external/tensorflow/tensorflow/c/eager/
Dtape.h450 gtl::FlatMap<int64, std::vector<Gradient*>> gradients; in ComputeGradient() local
/external/deqp/modules/gles3/functional/
Des3fTextureSizeTests.cpp229 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local
/external/deqp/modules/gles2/functional/
Des2fTextureSizeTests.cpp228 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local
Des2fTextureFilteringTests.cpp444 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/skia/src/pdf/
DSkPDFGradientShader.cpp162 int gradients = 0; in gradient_function_code() local
/external/skqp/src/pdf/
DSkPDFGradientShader.cpp162 int gradients = 0; in gradient_function_code() local
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py41 from tensorflow.python.ops import gradients_impl as gradients unknown
/external/deqp/modules/gles2/accuracy/
Des2aTextureFilteringTests.cpp412 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local
/external/deqp/modules/gles3/accuracy/
Des3aTextureFilteringTests.cpp398 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in init() local
/external/deqp/external/vulkancts/modules/vulkan/texture/
DvktTextureShadowTests.cpp453 static const tcu::Vec4 gradients[tcu::CUBEFACE_LAST][2] = in TextureCubeShadowTestInstance() local

12