| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| one_hot_op.cc | 46 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(depth_shape),
49 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value_shape),
52 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(off_value_shape),
|
| training_ops.cc | 40 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape),
81 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
92 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(momentum_shape),
139 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
184 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1_power_shape),
187 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta2_power_shape),
190 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
193 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1_shape),
196 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta2_shape),
199 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(epsilon_shape)
[all...] |
| matrix_band_part_op.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_lower_in_shape),
43 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_upper_in_shape),
|
| index_ops.cc | 41 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(dimension_shape),
|
| pad_op.cc | 75 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(ctx->InputShape(2)),
|
| select_op.cc | 50 bool cond_is_scalar = TensorShapeUtils::IsScalar(cond_shape);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| string_join_op.cc | 46 is_scalar.push_back(TensorShapeUtils::IsScalar(input.shape()));
47 if (!TensorShapeUtils::IsScalar(input.shape())) {
48 if (TensorShapeUtils::IsScalar(input_shape)) {
|
| one_hot_op.cc | 65 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(depth.shape()),
68 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value.shape()),
71 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(off_value.shape()),
|
| whole_file_read_ops.cc | 104 OP_REQUIRES(context, TensorShapeUtils::IsScalar(input->shape()),
128 OP_REQUIRES(context, TensorShapeUtils::IsScalar(filename_input->shape()),
132 OP_REQUIRES(context, TensorShapeUtils::IsScalar(contents_input->shape()),
|
| count_up_to_op.cc | 36 OP_REQUIRES(context, TensorShapeUtils::IsScalar(tensor.shape()),
76 context, TensorShapeUtils::IsScalar(before_increment.shape()),
|
| conditional_accumulator_base_op.cc | 41 if (!TensorShapeUtils::IsScalar(new_global_step_tensor->shape())) {
|
| encode_wav_op.cc | 40 OP_REQUIRES(context, TensorShapeUtils::IsScalar(sample_rate_tensor.shape()),
|
| extract_jpeg_shape_op.cc | 41 OP_REQUIRES(context, TensorShapeUtils::IsScalar(contents.shape()),
|
| matching_files_op.cc | 38 TensorShapeUtils::IsScalar(patterns_t->shape()) ||
|
| training_ops.cc | 589 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr.shape()),
593 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(rho.shape()),
597 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(epsilon.shape()),
736 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr.shape()),
740 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(rho.shape()),
744 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(epsilon.shape()),
[all...] |
| conditional_accumulator_base_op.h | 185 if (!TensorShapeUtils::IsScalar(local_step_tensor->shape())) {
222 if (!TensorShapeUtils::IsScalar(num_required_tensor->shape())) {
|
| generate_vocab_remapping_op.cc | 53 TensorShapeUtils::IsScalar(new_vocab_file_tensor->shape()),
85 TensorShapeUtils::IsScalar(old_vocab_file_tensor->shape()),
|
| sparse_to_dense_op.cc | 85 OP_REQUIRES(c, TensorShapeUtils::IsScalar(default_value.shape()),
110 if (TensorShapeUtils::IsScalar(sparse_values.shape())) {
|
| unravel_index_op.cc | 44 TensorShapeUtils::IsScalar(indices_tensor.shape()),
82 if (TensorShapeUtils::IsScalar(indices_tensor.shape())) {
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| dataset_utils.cc | 32 TensorShapeUtils::IsScalar(return_values[0].shape()))) {
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| decode_audio_op.cc | 119 OP_REQUIRES(context, TensorShapeUtils::IsScalar(contents_tensor.shape()),
123 OP_REQUIRES(context, TensorShapeUtils::IsScalar(file_format_tensor.shape()),
128 TensorShapeUtils::IsScalar(samples_per_second_tensor.shape()),
133 TensorShapeUtils::IsScalar(channel_count_tensor.shape()),
244 context, TensorShapeUtils::IsScalar(contents.shape()),
|
| encode_audio_op.cc | 82 OP_REQUIRES(context, TensorShapeUtils::IsScalar(file_format_tensor.shape()),
87 TensorShapeUtils::IsScalar(samples_per_second_tensor.shape()),
92 TensorShapeUtils::IsScalar(bits_per_second_tensor.shape()),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_query.cc | 104 return instruction->IsConstant() && ShapeUtil::IsScalar(instruction->shape());
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| rendezvous_util_test.cc | 43 CHECK(TensorShapeUtils::IsScalar(tensor.shape()));
|
| /external/tensorflow/tensorflow/stream_executor/ |
| device_memory.h | 127 bool IsScalar() const { return ElementCount() == 1; }
174 bool IsScalar() const { return ElementCount() == 1; }
|
