Home
last modified time | relevance | path

Searched refs:IsScalar (Results 1 – 25 of 139) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc40 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape), in Compile()
81 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape), in Compile()
92 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(momentum_shape), in Compile()
139 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape), in Compile()
184 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1_power_shape), in Compile()
187 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta2_power_shape), in Compile()
190 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape), in Compile()
193 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1_shape), in Compile()
196 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta2_shape), in Compile()
199 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(epsilon_shape), in Compile()
[all …]
Done_hot_op.cc46 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(depth_shape), in Compile()
49 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value_shape), in Compile()
52 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(off_value_shape), in Compile()
Dsequence_ops.cc155 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(start_in_shape), in Compile()
158 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(stop_in_shape), in Compile()
161 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(num_in_shape), in Compile()
Dmatrix_band_part_op.cc38 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_lower_in_shape), in Compile()
43 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_upper_in_shape), in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops.cc589 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr.shape()), in DoValidate()
593 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(rho.shape()), in DoValidate()
597 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(epsilon.shape()), in DoValidate()
736 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr.shape()), in DoCompute()
740 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(rho.shape()), in DoCompute()
744 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(epsilon.shape()), in DoCompute()
857 ctx, TensorShapeUtils::IsScalar(l1.shape()), in Compute()
862 ctx, TensorShapeUtils::IsScalar(l2.shape()), in Compute()
924 ctx, TensorShapeUtils::IsScalar(l1.shape()), in Compute()
929 ctx, TensorShapeUtils::IsScalar(l2.shape()), in Compute()
[all …]
Dstring_join_op.cc46 is_scalar.push_back(TensorShapeUtils::IsScalar(input.shape())); in Compute()
47 if (!TensorShapeUtils::IsScalar(input.shape())) { in Compute()
48 if (TensorShapeUtils::IsScalar(input_shape)) { in Compute()
Dwhole_file_read_ops.cc104 OP_REQUIRES(context, TensorShapeUtils::IsScalar(input->shape()), in Compute()
128 OP_REQUIRES(context, TensorShapeUtils::IsScalar(filename_input->shape()), in Compute()
132 OP_REQUIRES(context, TensorShapeUtils::IsScalar(contents_input->shape()), in Compute()
Done_hot_op.cc65 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(depth.shape()), in Compute()
68 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value.shape()), in Compute()
71 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(off_value.shape()), in Compute()
Dsequence_ops.cc128 OP_REQUIRES(context, TensorShapeUtils::IsScalar(start_in.shape()), in Compute()
131 OP_REQUIRES(context, TensorShapeUtils::IsScalar(stop_in.shape()), in Compute()
134 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_in.shape()), in Compute()
Dnon_max_suppression_op.cc163 context, TensorShapeUtils::IsScalar(max_output_size.shape()), in Compute()
189 context, TensorShapeUtils::IsScalar(max_output_size.shape()), in Compute()
194 OP_REQUIRES(context, TensorShapeUtils::IsScalar(iou_threshold.shape()), in Compute()
Dcount_up_to_op.cc36 OP_REQUIRES(context, TensorShapeUtils::IsScalar(tensor.shape()), in Compute()
76 context, TensorShapeUtils::IsScalar(before_increment.shape()), in Compute()
Dparameterized_truncated_normal_op.cc315 TensorShapeUtils::IsScalar(means_tensor.shape()) || in Compute()
323 TensorShapeUtils::IsScalar(stddevs_tensor.shape()) || in Compute()
331 TensorShapeUtils::IsScalar(minvals_tensor.shape()) || in Compute()
339 TensorShapeUtils::IsScalar(maxvals_tensor.shape()) || in Compute()
Dunravel_index_op.cc44 TensorShapeUtils::IsScalar(indices_tensor.shape()), in Compute()
82 if (TensorShapeUtils::IsScalar(indices_tensor.shape())) { in Compute()
Dexample_parsing_ops.cc185 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(serialized->shape()), in Compute()
312 TensorShapeUtils::IsScalar(context_dense_keys[di].shape()), in Compute()
321 TensorShapeUtils::IsScalar(context_sparse_keys[di].shape()), in Compute()
330 ctx, TensorShapeUtils::IsScalar(feature_list_dense_keys[di].shape()), in Compute()
340 ctx, TensorShapeUtils::IsScalar(feature_list_sparse_keys[di].shape()), in Compute()
366 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(debug_name->shape()), in Compute()
373 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(serialized->shape()), in Compute()
Dsparse_to_dense_op.cc85 OP_REQUIRES(c, TensorShapeUtils::IsScalar(default_value.shape()), in Compute()
110 if (TensorShapeUtils::IsScalar(sparse_values.shape())) { in Compute()
Dadjust_contrast_op.cc58 OP_REQUIRES(context, TensorShapeUtils::IsScalar(factor.shape()), in Compute()
61 OP_REQUIRES(context, TensorShapeUtils::IsScalar(min_value.shape()), in Compute()
64 OP_REQUIRES(context, TensorShapeUtils::IsScalar(max_value.shape()), in Compute()
166 OP_REQUIRES(context, TensorShapeUtils::IsScalar(factor.shape()), in Compute()
Dgenerate_vocab_remapping_op.cc53 TensorShapeUtils::IsScalar(new_vocab_file_tensor->shape()), in Compute()
85 TensorShapeUtils::IsScalar(old_vocab_file_tensor->shape()), in Compute()
/external/tensorflow/tensorflow/contrib/ffmpeg/
Ddecode_audio_op.cc119 OP_REQUIRES(context, TensorShapeUtils::IsScalar(contents_tensor.shape()), in Compute()
123 OP_REQUIRES(context, TensorShapeUtils::IsScalar(file_format_tensor.shape()), in Compute()
128 TensorShapeUtils::IsScalar(samples_per_second_tensor.shape()), in Compute()
133 TensorShapeUtils::IsScalar(channel_count_tensor.shape()), in Compute()
244 context, TensorShapeUtils::IsScalar(contents.shape()), in Compute()
Dencode_audio_op.cc82 OP_REQUIRES(context, TensorShapeUtils::IsScalar(file_format_tensor.shape()), in Compute()
87 TensorShapeUtils::IsScalar(samples_per_second_tensor.shape()), in Compute()
92 TensorShapeUtils::IsScalar(bits_per_second_tensor.shape()), in Compute()
/external/flatbuffers/src/
Didl_gen_general.cpp249 && !IsScalar(DestinationType(t, true).base_type) ? " @Nullable ": ""; in GenNullableAnnotation()
312 return IsScalar(type.base_type) in GenTypeGet()
438 return IsScalar(type.base_type) ? SourceCast(type, castFromDest) : ""; in SourceCastBasic()
496 if (!IsScalar(value.type.base_type)) { in GenDefaultValueBasic()
635 if (IsScalar(type.base_type)) { in GenSetter()
650 return IsScalar(type.base_type) in GenMethod()
929 (IsScalar(field.value.type.base_type) || in GenStruct()
931 IsScalar(field.value.type.element)))) { in GenStruct()
943 if (IsScalar(field.value.type.base_type)) { in GenStruct()
1008 (IsScalar(field.value.type.element) ? default_cast + "0" : "null"); in GenStruct()
[all …]
/external/tensorflow/tensorflow/contrib/image/kernels/
Dadjust_hsv_in_yiq_op.cc58 OP_REQUIRES(context, TensorShapeUtils::IsScalar(delta_h.shape()), in Compute()
61 OP_REQUIRES(context, TensorShapeUtils::IsScalar(scale_s.shape()), in Compute()
64 OP_REQUIRES(context, TensorShapeUtils::IsScalar(scale_v.shape()), in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover.cc75 if (ShapeUtil::IsScalar(instruction->shape())) { in CanTriviallyChangeShape()
100 if (!ShapeUtil::IsScalar(operand->shape()) && in FirstNonScalarAndNonTrivialReshapeOperand()
270 if (ShapeUtil::IsScalar(operands[i]->shape())) { in TrySinkReshapeOrTranspose()
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dclustering_ops.cc91 TensorShapeUtils::IsScalar(num_to_sample_tensor.shape()), in Compute()
93 OP_REQUIRES(context, TensorShapeUtils::IsScalar(seed_tensor.shape()), in Compute()
97 TensorShapeUtils::IsScalar(num_retries_per_sample_tensor.shape()), in Compute()
241 OP_REQUIRES(context, TensorShapeUtils::IsScalar(seed_tensor.shape()), in Compute()
298 OP_REQUIRES(context, TensorShapeUtils::IsScalar(k_tensor.shape()), in Compute()
Dwals_solver_ops.cc86 context, TensorShapeUtils::IsScalar(unobserved_weights.shape()), in Compute()
94 OP_REQUIRES(context, TensorShapeUtils::IsScalar(input_block_size.shape()), in Compute()
97 context, TensorShapeUtils::IsScalar(input_is_transpose.shape()), in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_memory.h127 bool IsScalar() const { return ElementCount() == 1; } in IsScalar() function
174 bool IsScalar() const { return ElementCount() == 1; } in IsScalar() function

123456