| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| D | categorical_split_handler_test.py | 35 def get_empty_tensors(gradient_shape, hessian_shape): argument
37 empty_grad_shape = [1] + gradient_shape.as_list()
63 gradient_shape = tensor_shape.scalar()
74 gradient_shape=gradient_shape,
81 gradient_shape, hessian_shape)
186 gradient_shape = tensor_shape.scalar()
197 gradient_shape=gradient_shape,
205 gradient_shape, hessian_shape)
297 gradient_shape = tensor_shape.scalar()
308 gradient_shape=gradient_shape,
[all …]
|
| D | ordinal_split_handler_test.py | 38 def get_empty_tensors(gradient_shape, hessian_shape): argument
40 empty_grad_shape = [1] + gradient_shape.as_list()
66 gradient_shape = tensor_shape.scalar()
78 gradient_shape=gradient_shape,
84 gradient_shape, hessian_shape)
200 gradient_shape = tensor_shape.scalar()
212 gradient_shape=gradient_shape,
219 gradient_shape, hessian_shape)
336 gradient_shape = tensor_shape.scalar()
348 gradient_shape=gradient_shape,
[all …]
|
| D | ordinal_split_handler.py | 100 gradient_shape, argument
134 gradient_shape=gradient_shape,
140 gradient_shape,
170 gradient_shape, argument
210 gradient_shape=gradient_shape,
353 gradient_shape, argument
389 gradient_shape=gradient_shape,
|
| D | categorical_split_handler.py | 45 gradient_shape, argument
80 gradient_shape=gradient_shape,
87 gradient_shape,
|
| D | base_split_handler.py | 39 gradient_shape, argument
69 self._gradient_shape = gradient_shape
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 86 StatsAccumulatorResource(const TensorShape& gradient_shape, in StatsAccumulatorResource() argument
88 : gradient_shape_(gradient_shape), in StatsAccumulatorResource()
94 TensorShapeUtils::IsScalar(gradient_shape)); in StatsAccumulatorResource()
114 const TensorShape& gradient_shape() const { return gradient_shape_; } in gradient_shape() function in tensorflow::boosted_trees::__anon2081c9750111::StatsAccumulatorResource
189 TensorShape gradient_shape = accumulator_resource.gradient_shape(); in SerializeTensorAccumulatorToOutput() local
190 int64 num_gradient_elements = gradient_shape.num_elements(); in SerializeTensorAccumulatorToOutput()
191 gradient_shape.InsertDim(0, num_slots); in SerializeTensorAccumulatorToOutput()
194 context->allocate_output("output_gradients", gradient_shape, in SerializeTensorAccumulatorToOutput()
286 context, gradients_shape == accumulator_resource->gradient_shape(), in AddToTensorAccumulator()
289 ", ", accumulator_resource->gradient_shape().DebugString()))); in AddToTensorAccumulator()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | stats_accumulator_ops_test.py | 35 gradient_shape=tensor_shape.scalar(),
63 gradient_shape=tensor_shape.scalar(),
92 gradient_shape=tensor_shape.scalar(),
124 gradient_shape=tensor_shape.scalar(),
165 gradient_shape=tensor_shape.scalar(),
202 gradient_shape=tensor_shape.scalar(),
224 gradient_shape=tensor_shape.TensorShape([2]),
262 gradient_shape=tensor_shape.TensorShape([2]),
300 gradient_shape=tensor_shape.TensorShape([2]),
337 gradient_shape=tensor_shape.TensorShape([2]),
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseAccumulatorApplyGradient.pbtxt | 31 name: "gradient_shape"
46 Boolean indicating whether gradient_shape is unknown, in which
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | stats_accumulator_ops.py | 116 gradient_shape, argument
133 self._gradient_shape = gradient_shape
137 if (gradient_shape == tensor_shape.scalar() and
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 157 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local
170 gradient_shape.dim_sizes()); in Compile()
240 const TensorShape gradient_shape = ctx->InputShape(0); in Compile() local
256 xla::XlaOp zeroes = xla::Broadcast(zero, gradient_shape.dim_sizes()); in Compile()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | optimizers.py | 421 gradient_shape = gradient.dense_shape
423 gradient_shape = gradient.get_shape()
424 noise = random_ops.truncated_normal(gradient_shape) * gradient_noise_scale
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 744 gradient_shape=self._gradient_shape,
771 gradient_shape=self._gradient_shape,
794 gradient_shape=self._gradient_shape,
831 gradient_shape=self._gradient_shape,
837 gradient_shape=tensor_shape.scalar(),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1432 gradient_shape=math_ops.cast(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3409 …ndle\', \'local_step\', \'gradient_indices\', \'gradient_values\', \'gradient_shape\', \'has_known…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3409 …ndle\', \'local_step\', \'gradient_indices\', \'gradient_values\', \'gradient_shape\', \'has_known…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 70773 name: "gradient_shape"
70823 name: "gradient_shape"
70875 name: "gradient_shape"
70928 name: "gradient_shape"
|
| D | ops_history.v1.pbtxt | 71400 name: "gradient_shape"
71450 name: "gradient_shape"
71502 name: "gradient_shape"
71555 name: "gradient_shape"
|
| D | ops_history.v0.pbtxt | 31855 name: "gradient_shape"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 35086 name: "gradient_shape"
|