| /external/tensorflow/tensorflow/core/framework/ |
| tensor_shape.cc | 548 bool TensorShapeUtils::StartsWith(const TensorShape& shape,
557 bool TensorShapeUtils::EndsWith(const TensorShape& shape,
610 Status TensorShapeUtils::MakeShape(const T* dims, int64 n, Shape* out) { \
613 Status TensorShapeUtils::MakeShape(gtl::ArraySlice<T> shape, Shape* out) { \
622 string TensorShapeUtils::ShapeListString(
755 Status TensorShapeUtils::NumElements(gtl::ArraySlice<int64> shape,
|
| tensor_shape.h | 270 // For use by TensorShapeUtils::MakeShape
354 class TensorShapeUtils {
444 return TensorShapeUtils::MakeShape(dims, n, out);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| set_kernels.cc | 264 TensorShapeUtils::MakeShape(output_shape, &output_shape_ts));
450 TensorShapeUtils::NumElements(group_shape, &num_elements));
472 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(group_shape, &output_shape));
509 TensorShapeUtils::NumElements(group_shape, &num_elements));
554 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(group_shape, &output_shape));
677 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(group_shape, &output_shape));
|
| segment_reduction_ops.cc | 57 OP_REQUIRES(context, TensorShapeUtils::IsVector(segment_ids.shape()),
225 context, TensorShapeUtils::IsVector(segment_ids.shape()),
446 context, TensorShapeUtils::StartsWith(data.shape(), segment_ids.shape()),
648 OP_REQUIRES(context, TensorShapeUtils::IsVector(indices.shape()),
650 OP_REQUIRES(context, TensorShapeUtils::IsVector(segment_ids.shape()),
[all...] |
| tensor_array_ops.cc | 187 if (!TensorShapeUtils::IsScalar(tensor_size->shape())) {
347 TF_RETURN_IF_ERROR(TensorShapeUtils::MakeShape(
429 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_index->shape()),
512 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_index->shape()),
629 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(tensor_indices->shape()),
[all...] |
| random_op.cc | 227 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(minval.shape()),
230 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(maxval.shape()),
279 TensorShapeUtils::IsVector(shape_t.shape()) &&
288 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(vec.data(), vec.size(),
292 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(vec.data(), vec.size(),
[all...] |
| barrier_ops.cc | 487 TensorShapeUtils::ShapeListString(barrier->component_shapes()),
489 TensorShapeUtils::ShapeListString(value_component_shapes_));
586 OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(Tnum_elements->shape()),
|
| functional_ops.cc | 65 if (TensorShapeUtils::IsScalar(t[0].shape())) {
244 OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(branch_index.shape()),
490 if (!TensorShapeUtils::IsScalar(t.shape())) {
|
| inplace_ops.cc | 262 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(i.shape()),
440 ctx, TensorShapeUtils::IsVector(shape.shape()),
445 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(
|
| mirror_pad_op.cc | 74 TensorShapeUtils::IsMatrix(in1.shape()) && in1.dim_size(1) == 2,
285 TensorShapeUtils::IsMatrix(in1.shape()) && in1.dim_size(1) == 2,
|
| multinomial_op.cc | 155 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_t.shape()),
159 ctx, TensorShapeUtils::IsScalar(num_samples_t.shape()),
|
| adjust_hue_op.cc | 60 OP_REQUIRES(context, TensorShapeUtils::IsScalar(delta.shape()),
|
| adjust_saturation_op.cc | 59 OP_REQUIRES(context, TensorShapeUtils::IsScalar(scale.shape()),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| image_ops.cc | 185 OP_REQUIRES(context, TensorShapeUtils::IsScalar(factor_shape),
228 OP_REQUIRES(context, TensorShapeUtils::IsScalar(scale_shape),
282 OP_REQUIRES(context, TensorShapeUtils::IsScalar(delta_shape),
408 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(boxes_shape),
416 OP_REQUIRES(context, TensorShapeUtils::IsVector(scores_shape),
|
| dynamic_stitch_op.cc | 77 ctx, TensorShapeUtils::StartsWith(data_shape, indices_shape),
|
| split_op.cc | 42 ctx, TensorShapeUtils::IsScalar(split_dim_shape),
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| bigtable_kernels.cc | 323 if (!TensorShapeUtils::IsScalar(tensors[i].shape())) {
345 if (!TensorShapeUtils::IsScalar(argument_t->shape())) {
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| interleave_dataset_op.cc | 44 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(cycle_length_t->shape()),
53 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(block_length_t->shape()),
|
| padded_batch_dataset_op.cc | 62 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(padded_shape_t.shape()),
84 ctx, TensorShapeUtils::IsScalar(padding_value_t.shape()),
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| xrt_compile_ops.cc | 148 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(computation_input.shape()),
|
| /external/tensorflow/tensorflow/contrib/cloud/kernels/ |
| gcs_config_ops.cc | 70 if (!TensorShapeUtils::IsScalar(argument_t->shape())) {
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| pmf_to_cdf_op.cc | 52 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(shape),
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| sparse_feature_cross_kernel.cc | 403 context, TensorShapeUtils::IsMatrix(indices_list_in[i].shape()),
421 context, TensorShapeUtils::IsVector(values_list_in[i].shape()),
443 context, TensorShapeUtils::IsVector(shapes_list_in[i].shape()),
462 context, TensorShapeUtils::IsMatrix(dense_list_in[i].shape()),
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| rpc_rendezvous_mgr_test.cc | 39 CHECK(TensorShapeUtils::IsScalar(tensor.shape()));
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| rebatch.cc | 155 if (!TensorShapeUtils::IsScalar(batch_size_tensor.shape())) {
|
