Searched refs:cond_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | select_op.cc | 34 const TensorShape cond_shape = ctx->InputShape(0); in Compile() local
48 bool broadcasting = !cond_shape.IsSameSize(then_shape); in Compile()
49 bool cond_is_scalar = TensorShapeUtils::IsScalar(cond_shape); in Compile()
51 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(cond_shape), in Compile()
54 cond_shape.DebugString())); in Compile()
59 OP_REQUIRES(ctx, then_shape.dim_size(0) == cond_shape.num_elements(), in Compile()
63 cond_shape.num_elements())); in Compile()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | where.cc | 36 const RuntimeShape& cond_shape = GetTensorShape(cond_tensor); in ResizeOutputTensor() local
37 const int size = cond_shape.FlatSize(); in ResizeOutputTensor()
38 const int cond_rank = cond_shape.DimensionsCount(); in ResizeOutputTensor()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 412 for cond_shape in [2], [2, 3], [2, 3, 5]:
413 cond = random_ops.random_uniform(cond_shape) > 0.5
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 3707 cond_shape = array_ops.concat(
3709 condition = array_ops.reshape(condition, cond_shape)
3711 shape_diff = expr_shape - cond_shape
|