Home
last modified time | relevance | path

Searched defs:gradient_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dcategorical_split_handler_test.py35 def get_empty_tensors(gradient_shape, hessian_shape): argument
Dordinal_split_handler_test.py38 def get_empty_tensors(gradient_shape, hessian_shape): argument
Dcategorical_split_handler.py45 gradient_shape, argument
Dordinal_split_handler.py100 gradient_shape, argument
170 gradient_shape, argument
353 gradient_shape, argument
Dbase_split_handler.py39 gradient_shape, argument
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc86 StatsAccumulatorResource(const TensorShape& gradient_shape, in StatsAccumulatorResource()
114 const TensorShape& gradient_shape() const { return gradient_shape_; } in gradient_shape() function in tensorflow::boosted_trees::__anon539a428c0111::StatsAccumulatorResource
189 TensorShape gradient_shape = accumulator_resource.gradient_shape(); in SerializeTensorAccumulatorToOutput() local
364 TensorShape gradient_shape = TensorShape({}); in Compute() local
398 TensorShape gradient_shape = TensorShape(gradient_shape_t->vec<int64>()); in Compute() local
752 TensorShape gradient_shape = TensorShape({}); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_quantize_ops.cc157 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local
240 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py116 gradient_shape, argument