Searched defs:gradient_shape (Results 1 – 8 of 8) sorted by relevance
35 def get_empty_tensors(gradient_shape, hessian_shape): argument
38 def get_empty_tensors(gradient_shape, hessian_shape): argument
45 gradient_shape, argument
100 gradient_shape, argument170 gradient_shape, argument353 gradient_shape, argument
39 gradient_shape, argument
86 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::StatsAccumulatorResource189 TensorShape gradient_shape = accumulator_resource.gradient_shape(); in SerializeTensorAccumulatorToOutput() local364 TensorShape gradient_shape = TensorShape({}); in Compute() local398 TensorShape gradient_shape = TensorShape(gradient_shape_t->vec<int64>()); in Compute() local752 TensorShape gradient_shape = TensorShape({}); in Compute() local
157 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local240 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local
116 gradient_shape, argument