Home
last modified time | relevance | path

Searched refs:feature_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dboosted_trees_ops.cc193 shape_inference::ShapeHandle feature_shape; in __anone3ce78ae0702() local
199 TF_RETURN_IF_ERROR(c->WithRank(c->input(i + 1), 1, &feature_shape)); in __anone3ce78ae0702()
201 TF_RETURN_IF_ERROR(c->Merge(c->input(1), feature_shape, &unused_input)); in __anone3ce78ae0702()
207 c->MakeShape({c->Dim(feature_shape, 0), logits_dimension}); in __anone3ce78ae0702()
220 shape_inference::ShapeHandle feature_shape; in __anone3ce78ae0802() local
226 TF_RETURN_IF_ERROR(c->WithRank(c->input(i + 1), 1, &feature_shape)); in __anone3ce78ae0802()
228 TF_RETURN_IF_ERROR(c->Merge(c->input(1), feature_shape, &unused_input)); in __anone3ce78ae0802()
232 auto batch_size = c->MakeShape({c->Dim(feature_shape, 0)}); in __anone3ce78ae0802()
260 shape_inference::ShapeHandle feature_shape; in __anone3ce78ae0a02() local
267 TF_RETURN_IF_ERROR(c->WithRank(c->input(i + 3), 1, &feature_shape)); in __anone3ce78ae0a02()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc212 const Shape feature_shape = scale->shape(); in HandleBatchNormTraining() local
247 auto sum = add(HloInstruction::CreateReduce(feature_shape, operand, zero, in HandleBatchNormTraining()
253 feature_shape, operand_squared, zero, dimensions_without_feature, in HandleBatchNormTraining()
267 add_binary(feature_shape, HloOpcode::kMultiply, mean, mean); in HandleBatchNormTraining()
271 add_binary(feature_shape, HloOpcode::kSubtract, square_mean, mean_square); in HandleBatchNormTraining()
341 const Shape feature_shape = scale->shape(); in HandleBatchNormInference() local
465 const Shape feature_shape = scale->shape(); in HandleBatchNormGrad() local
486 add(HloInstruction::CreateBroadcast(feature_shape, epsilon_scalar, {})); in HandleBatchNormGrad()
512 add_binary(feature_shape, HloOpcode::kAdd, variance, epsilon_feature), in HandleBatchNormGrad()
530 feature_shape, grad_output_times_activation_minus_mean, zero, in HandleBatchNormGrad()
[all …]
Dshape_inference.cc1527 Shape feature_shape = in InferBatchNormGradShape() local
1568 {operand_shape, feature_shape, feature_shape}); in InferBatchNormGradShape()
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead.py380 feature_shape = value.get_shape()
381 if feature_shape.ndims is None:
383 if feature_shape.ndims < 2:
386 "(got rank {} for feature '{}')").format(feature_shape.ndims, name))
387 if not feature_shape[:2].is_compatible_with(
396 feature_shape=feature_shape,
Destimators.py275 feature_shape = ([default_batch_size, default_series_length]
278 dtype=value_dtype, name=feature_key, shape=feature_shape)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py332 def create_dummy_tensor(feature_shape, feature_type): argument
341 for dim in feature_shape.dims:
346 if feature_shape.is_fully_defined() and dims:
355 for feature_shape, feature_type in zip(value_structure._flat_shapes,
357 result.append(create_dummy_tensor(feature_shape, feature_type))