Searched refs:alpha_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 39 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local
40 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape), in Compile()
42 alpha_shape.DebugString())); in Compile()
84 TensorShape alpha_shape = ctx->InputShape(1); in Compile() local
85 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape), in Compile()
87 alpha_shape.DebugString())); in Compile()
89 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape), in Compile()
93 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape), in Compile()
945 TensorShape alpha_shape = ctx->InputShape(3); in CheckScalarParams() local
946 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape), in CheckScalarParams()
[all …]
|
| /external/tensorflow/tensorflow/contrib/gan/python/losses/python/ |
| D | losses_impl.py | 363 alpha_shape = [batch_size] + [1] * (differences.shape.ndims - 1)
364 alpha = random_ops.random_uniform(shape=alpha_shape)
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 821 alpha_shape = []
823 alpha_shape.append(1 if dim in shared_axes else input_shape[dim])
825 alpha_values = create_tensor_data(np.float32, alpha_shape)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 4603 const RuntimeShape& alpha_shape, in BroadcastPrelu4DSlow() argument
4608 TFLITE_DCHECK_LE(alpha_shape.DimensionsCount(), 4); in BroadcastPrelu4DSlow()
4614 NdArrayDescsForElementwiseBroadcast(input_shape, alpha_shape, &desc1, &desc2); in BroadcastPrelu4DSlow()
|