Home
last modified time | relevance | path

Searched refs:flat_outer_dims (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc581 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local
582 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F()
583 EXPECT_EQ(60, flat_outer_dims.dimension(1)); in TEST_F()
584 EXPECT_EQ(flat_outer_dims(0, 0), 0.01f); in TEST_F()
585 EXPECT_EQ(flat_outer_dims(1, 59), 0.02f); in TEST_F()
588 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local
589 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F()
590 EXPECT_EQ(3, flat_outer_dims.dimension(1)); in TEST_F()
591 EXPECT_EQ(20, flat_outer_dims.dimension(2)); in TEST_F()
592 EXPECT_EQ(flat_outer_dims(0, 0, 0), 0.01f); in TEST_F()
[all …]
Dtensor.h481 typename TTypes<T, NDIMS>::Tensor flat_outer_dims();
582 typename TTypes<T, NDIMS>::ConstTensor flat_outer_dims() const;
918 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function
935 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function
/external/tensorflow/tensorflow/core/kernels/
Drandom_shuffle_op.cc93 const auto input_mat = input.flat_outer_dims<T>(); in Compute()
94 auto output_mat = output->flat_outer_dims<T>(); in Compute()
Dsegment_reduction_ops_impl.h100 auto input_flat = input.flat_outer_dims<T>(); in Compute()
123 auto output_flat = output->flat_outer_dims<T>(); in Compute()
295 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync()
414 auto output_flat = output->flat_outer_dims<T>();
476 auto input_flat = input.flat_outer_dims<T>();
505 output->flat_outer_dims<T>().setConstant(default_value_);
511 auto output_flat = output->flat_outer_dims<T>();
517 auto temp_flat = temp.flat_outer_dims<float>();
864 auto input_flat = input.flat_outer_dims<T>();
900 auto output_flat = output->flat_outer_dims<T>();
Dinplace_ops_functor_gpu.cu.cc48 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
113 auto Ty = y->flat_outer_dims<T>(); in DoInplaceOp()
148 auto Ty = y->flat_outer_dims<bool>(); in DoInplaceOp()
Dinplace_ops.cc35 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
247 auto Tv = v.flat_outer_dims<T>(); in DoInplaceOp()
248 auto Ty = y->flat_outer_dims<T>(); in DoInplaceOp()
270 auto Tv = v.flat_outer_dims<tstring>(); in DoInplaceStringUpdateOp()
271 auto Ty = y->flat_outer_dims<tstring>(); in DoInplaceStringUpdateOp()
Dcwise_op_select.cc79 FastBoundsCheck(then->flat_outer_dims<T>().dimension(1), in ComputeBroadcasting()
105 func(ctx->eigen_device<Device>(), output->flat_outer_dims<T>(), in ComputeBroadcasting()
106 cond->vec<bool>(), then->flat_outer_dims<T>(), in ComputeBroadcasting()
107 else_->flat_outer_dims<T>()); in ComputeBroadcasting()
Dsparse_conditional_accumulator.h264 auto sum_flat = sum_tensor->flat_outer_dims<T>(); in AddToAccumGradFunction()
265 auto accum_flat = accum_val_->flat_outer_dims<T>(); in AddToAccumGradFunction()
266 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction()
324 auto accum_flat = accum_val_->flat_outer_dims<T>(); in DivideAccumGradByCounter()
Dtraining_ops.cc1260 auto var_flat = var.flat_outer_dims<T>(); in DoCompute()
1261 auto accum_grad_flat = accum_grad.flat_outer_dims<T>(); in DoCompute()
1262 auto accum_update_flat = accum_update.flat_outer_dims<T>(); in DoCompute()
1263 auto grad_flat = grad.flat_outer_dims<T>(); in DoCompute()
1444 auto var_flat = var.flat_outer_dims<T>(); in Compute()
1445 auto grad_flat = grad.flat_outer_dims<T>(); in Compute()
1913 device, var.flat_outer_dims<T>(), accum.flat_outer_dims<T>(), in Compute()
1915 lr.scalar<T>(), lr.scalar<T>(), grad.flat_outer_dims<T>(), in Compute()
2047 device, var.flat_outer_dims<T>(), accum.flat_outer_dims<T>(), in Compute()
2048 lr.scalar<T>(), epsilon.scalar<T>(), grad.flat_outer_dims<T>(), in Compute()
[all …]
Dragged_tensor_from_variant_op.cc175 output_ragged->mutable_values()->flat_outer_dims<VALUE_TYPE, 2>(); in NestedStackRaggedTensors()
179 ragged_components[i].values().flat_outer_dims<VALUE_TYPE, 2>(); in NestedStackRaggedTensors()
Dragged_gather_op.cc39 params_dense_values_in.flat_outer_dims<VALUE_TYPE, 2>(); in WriteValueSlices()
40 auto values = values_out->flat_outer_dims<VALUE_TYPE, 2>(); in WriteValueSlices()
Ddynamic_stitch_op.cc187 const int slice_size = merged->flat_outer_dims<T>().dimension(1); in Compute()
251 auto merged_flat = merged->flat_outer_dims<T>(); in Compute()
Dgather_nd_op.h136 auto params_flat = params.flat_outer_dims<T, IXDIM + 1>(); \ in DoGatherNd()
Ddynamic_partition_op.cc132 out_flat.push_back(outputs[p]->flat_outer_dims<T>()); in Compute()
Dscatter_op.cc119 auto params_flat = params.flat_outer_dims<T>(); in DoCompute()
Dqueue_base.cc45 auto parent_as_matrix = parent.flat_outer_dims<T>(); in HandleSliceToElement()
Dbarrier_ops.cc312 auto values_matrix = values.flat_outer_dims<T>(); in InsertOneLocked()
Dresource_variable_ops.cc905 auto params_flat = params->flat_outer_dims<T>(); in DoCompute()
Dlist_kernels.h986 auto input_t = input.flat_outer_dims<T, 2>(); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc695 auto output_t = output->flat_outer_dims<type>(); \ in CopyPartialBatch()
696 auto value_t = value.flat_outer_dims<type>(); \ in CopyPartialBatch()